MIC-DKFZ / batchgenerators

A framework for data augmentation for 2D and 3D image classification and segmentation
Apache License 2.0
1.09k stars 221 forks source link

Make transforms=None pass in STA #35

Closed justusschock closed 5 years ago

justusschock commented 5 years ago

Multithreaded augmenter accepts transforms to be None (for applying no transform at all), but singlethreaded augmenter fails on it.

This one-liner should fix this and make the behavior same for both augmenters.

cc @FabianIsensee

FabianIsensee commented 5 years ago

Hi, I just integrated your changes. Due to the repo being hosted internally I need to merge changes differently (not via github). But your authorship of the changes is of course retained. Best, Fabian