rotation augmentation

[astro-fits]

Hi,

A great work.
My question is , why shall we only perform rotation augmentation on large fg images ( width/height > 1024 pix) during training ?

https://github.com/Yaoyi-Li/GCA-Matting/blob/master/dataloader/data_generator.py