ELEKTRONN / elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data
MIT License
160 stars 27 forks source link

Add non-geometric augmentation methods #9

Closed mdraw closed 4 years ago

mdraw commented 6 years ago

Here are many examples of image augmentations. The geometric ones (rotation, scaling, affine etc.) are already implemented in elektronn3 and many of the others are not suitable for biomedical data sets, but some of them like "ContrastNormalization (per channel)", "Add/Multipy (per channel)" and "SaltAndPepper" could be really helpful for us. We can't use https://github.com/aleju/imgaug directly because it depends on OpenCV and is not (fully) 3D compatible, but implementing those methods in numpy for 3D/2D ourselves shouldn't be too hard.

mdraw commented 4 years ago

All of the mentioned non-geometric augmentations that still make sense are now supported in elektronn3.data.transforms, so this issue can be closed.