facebookresearch / unbiased-teacher

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection
https://arxiv.org/abs/2102.09480
MIT License
411 stars 82 forks source link

Random erasing fully erasing objects #58

Open guilhermecorreagv opened 2 years ago

guilhermecorreagv commented 2 years ago

Firstly thank you for your work, it is helping me improve my model through the use of unlabeled data.

I've copied the strong augmentation used in ubteacher.data.detection_utils, but in my dataset I have some small objects, and very frequently they get completely erased by the random erasing, and forgive me if I overlooked it, but I don't see anywhere in your repository a place where you check if objects are being erased.

Is this step done somewhere? Or did you use the random erasing without checking this?

Thank you in advance.