TF compatible image augmentations All of the functions defined here are compatible as mapping functions for tf.data.Dataset.map
The following modules are included:
- affine - affine transformations
- elastic - elastic deformation (inspired by the U-Net paper)
- binary mask - extract information from an object binary mask
- seg_aug - segmentation augmentations, random augmentations that preserve compatibility between an input image and it's segmentation mask
- utils - general utils, mostly involving dimensions.
If you're using my work for anything other than personal use remember to give credit to Sol Yarkoni.