lkeab / BCNet

Deep Occlusion-Aware Instance Segmentation with Overlapping BiLayers [CVPR 2021]
https://arxiv.org/abs/2103.12340
MIT License
529 stars 74 forks source link

data augmentation #142

Open CHN-001 opened 3 months ago

CHN-001 commented 3 months ago

Hello author, I would like to use online data augmentation in BCNet now. I see that the image size can be randomly processed in the cfg file,So I want to add some other enhancements, such as the ones shown below,

T.RandomBrightness(0.9, 1.1) and T.RandomFlip(prob=0.5)

and I'm not sure how to add them. Should I modify the datamapper.py file? Or should I add some code to my train.py file? I'm looking forward to your reply and i will be grateful if u can help me! Please forgive my poor English.

These are some related works that I found, and i hope these can help us. https://stackoverflow.com/questions/71774744/how-to-use-detectron2s-augmentation-with-datasets-loaded-using-register-coco-in/71799464#71799464 https://github.com/facebookresearch/detectron2/issues/1578