okankop / vidaug

Effective Video Augmentation Techniques for Training Convolutional Neural Networks
MIT License
387 stars 78 forks source link

RandomRotate does not preserve range #23

Open dermahax opened 1 year ago

dermahax commented 1 year ago

Hey,

when using RandomRotate with numpy arrays, the images are normalized. For most use cases it might be preferable to prevent the range of input imgs. Skikit.transform.rotate has the flag preserver_range (default False).

https://scikit-image.org/docs/stable/api/skimage.transform.html#skimage.transform.rotate

It could be a good idea to forward it to the user or set it to True in the va.RandomRotate class.

Cheers, Max.