본문 바로가기
  • Home
Basic Search

Small-Scale Object Detection Label Reassignment Strategy

  • Journal of The Korea Society of Computer and Information
  • Abbr : JKSCI
  • 2022, 27(12), pp.77-84
  • DOI : 10.9708/jksci.2022.27.12.077
  • Publisher : The Korean Society Of Computer And Information
  • Research Area : Engineering > Computer Science
  • Received : November 28, 2022
  • Accepted : December 23, 2022
  • Published : December 30, 2022

Jung-In An 1,  Kim, yoon 1,  Hyun-Soo Choi 2

1강원대학교
2서울과학기술대학교

Accredited

ABSTRACT

In this paper, we propose a Label Reassignment Strategy to improve the performance of an object detection algorithm. Our approach involves two stages: an inference stage and an assignment stage. In the inference stage, we perform multi-scale inference with predefined scale sizes on a trained model and re-infer masked images to obtain robust classification results. In the assignment stage, we calculate the IoU between bounding boxes to remove duplicates. We also check box and class occurrence between the detection result and annotation label to re-assign the dominant class type. We trained the YOLOX-L model with the re-annotated dataset to validate our strategy. The model achieved a 3.9% improvement in mAP and 3x better performance on AP_S compared to the model trained with the original dataset. Our results demonstrate that the proposed Label Reassignment Strategy can effectively improve the performance of an object detection model.

Statistics

  • 241 Viewed
  • 112 Downloaded

Tools

Issue List

Citation status

* References for papers published after 2023 are currently being built.

This paper was written with support from the National Research Foundation of Korea.
Basic Search Advanced Search