Waikato / wekaDeeplearning4j

Weka package for the Deeplearning4j java library
https://deeplearning.cms.waikato.ac.nz/
GNU General Public License v3.0
184 stars 197 forks source link

Default Image Data Augmentation #70

Open basedrhys opened 2 years ago

basedrhys commented 2 years ago

Is your feature request related to a problem? Please describe. A common technique when training a CNN is to perform data augmentation (crop/shear/rotate/etc.) to improve the model's generalisability. This currently doesn't exist in WDL4J

Describe the solution you'd like Some reasonable default augmentation methods to apply to the image data during training.

Additional context Fast.ai has some good examples