Improved Hard Example Mining Approach for Single Shot Object Detectors

Köksal A., Tuzcuoğlu Ö., Ince K. G., Ataseven Y., Alatan A. A.

2022 IEEE International Conference on Image Processing (ICIP), Bordeaux, Fransa, 16 Ekim 2022 identifier identifier

  • Yayın Türü: Bildiri / Tam Metin Bildiri
  • Doi Numarası: 10.1109/icip46576.2022.9897806
  • Basıldığı Şehir: Bordeaux
  • Basıldığı Ülke: Fransa
  • Orta Doğu Teknik Üniversitesi Adresli: Evet

Özet

Hard example mining methods generally improve the performance of the object detectors, which suffer from imbalanced training sets. In this work, two existing hard example mining approaches (LRM and focal loss, FL) are adapted and combined in a state-of-the-art real-time object detector, YOLOv5. The effectiveness of the proposed approach for improving the performance on hard examples is extensively evaluated. The proposed method increases mAP by 3% compared to using the original loss function and around 1-2% compared to using the hard-mining methods (LRM or FL) individually on 2021 Anti-UAV Challenge Dataset.