Back to Search Start Over

Mixed Label Assignment Realizes End-to-End Object Detection.

Authors :
Chen, Jiaquan
Shao, Changbin
Su, Zhen
Source :
Electronics (2079-9292); Dec2024, Vol. 13 Issue 23, p4856, 15p
Publication Year :
2024

Abstract

Currently, detectors have made significant progress in inference speed and accuracy. However, these detectors require Non-Maximum Suppression (NMS) during the post-processing stage to eliminate redundant boxes, which limits the optimization of model inference speed. We first analyzed the reason for the dependence on NMS in the post-processing stage. The result showed that a score loss in a one-to-many label assignment leads to the presence of high-quality redundant boxes, making them difficult to remove. To realize end-to-end object detection and simplify the detection pipeline, we propose herein a mixed label assignment (MLA) training method, which uses one-to-many label assignment to provide rich supervision signals, alleviating the performance degradation, and we eliminate the need for NMS in the post-processing stage by using one-to-one label assignment. Additionally, a window feature propagation block (WFPB) is introduced, utilizing the inductive bias of images to enable feature sharing in local regions. Through these methods, we conducted experiments on the VOC and DUO datasets; our end-to-end detector MA-YOLOX achieved 66.0 mAP and 52.6 mAP, respectively, outperforming the YOLOX by 1.7 and 1.6. Additionally, our model performed faster than other real-time detectors without NMS. [ABSTRACT FROM AUTHOR]

Subjects

Subjects :
DETECTORS
SPEED

Details

Language :
English
ISSN :
20799292
Volume :
13
Issue :
23
Database :
Complementary Index
Journal :
Electronics (2079-9292)
Publication Type :
Academic Journal
Accession number :
181654528
Full Text :
https://doi.org/10.3390/electronics13234856