How to switch ON/OFF data augmentation?

experiencor / keras-yolo2

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).

MIT License

1.73k stars 784 forks source link

How to switch ON/OFF data augmentation? #295

Open bexx1963 opened 6 years ago

bexx1963 commented 6 years ago

I can find in preprocessing.py that some data augmentation is done to the dataset. Is there any chance of disabling augmentation?

rodrigo2019 commented 6 years ago

in this line add the parameter jitter=False