search

microsoft / SoftTeacher

Semi-Supervised Learning, Object Detection, ICCV2021
MIT License
892 stars 123 forks source link

Training on Imbalanced Custom Data #215

Open Hananali1 opened 2 years ago

Hananali1 commented 2 years ago

Thanks a lot for this great work. I'm working on imbalanced custom data with two classes. Both the baseline model and SoftTeacher model detect only one class on the test set. What changes to the Faster RCNN model (faster_rcnn_r50_fpn.py) you would suggest mitigating the overfitting towards the major class? Thanks