Yaoyi-Li / GCA-Matting

Official repository for Natural Image Matting via Guided Contextual Attention
MIT License
390 stars 62 forks source link

rotation augmentation #30

Open astro-fits opened 10 months ago

astro-fits commented 10 months ago

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

Screen Shot 2023-10-23 at 13 23 40