anuragranj / spynet

Spatial Pyramid Network for Optical Flow
Other
236 stars 48 forks source link

Random rotation in training #21

Open fukka opened 1 year ago

fukka commented 1 year ago

Hi, excellent paper! I am reading your code. I see you have only random scale, random noise, random crop, and random color as data augmentation in donkey.lua. But you mentioned random rotation in your paper. Can you show an example of how to use it? I am mostly wondering how you handle the flow with random rotation.

anuragranj commented 1 year ago

https://github.com/anuragranj/spynet/blob/master/donkey.lua#L312-L313