• Optics and Precision Engineering
  • Vol. 30, Issue 13, 1620 (2022)
Shuzhi SU1,2,*, Runbin CHEN1, Yanmin ZHU3, and Bowen JIANG1
Author Affiliations
  • 1School of Computer Science and Engineering, Anhui University of Science & Technology, Huainan23200, China
  • 2Institute of Energy, Hefei Comprehensive National Science Center, Hefei30031, China
  • 3School of Computer Mechanical Engineering, Anhui University of Science & Technology, Huainan22001, China
  • show less
    DOI: 10.37188/OPE.20223013.1620 Cite this Article
    Shuzhi SU, Runbin CHEN, Yanmin ZHU, Bowen JIANG. Relocation non-maximum suppression algorithm[J]. Optics and Precision Engineering, 2022, 30(13): 1620 Copy Citation Text show less

    Abstract

    Non-Maximum Suppression (NMS) is a post-processing algorithm used for object detection. It selects optimal bounding boxes from the bounding boxes set and suppresses other bounding boxes. NMS selects the bounding box with the highest score of classification confidence as the optimal bounding box. However, it ignores the correlation between localization accuracy and the classification confidence score. The classification confidence score cannot effectively represent the localization accuracy. This paper proposes a novel Relocation Non-Maximum Suppression (R-NMS) algorithm to solve the above-mentioned problem. First, the bounding box with the highest score of classification confidence in the bounding boxes set is selected as the optimal bounding box. Second, a new box distance measurement method is proposed based on R-NMS instead of using Intersection over Union (IoU) to measure the distance between the bounding boxes. Then, the location information of the bounding boxes around the optimal bounding box is obtained. Finally, the location information is used to relocate the optimal bounding box to obtain the new optimal bounding box. Compared with NMS and Soft-NMS, the mAP of R-NMS on YOLOv3 increased by 0.7 % and 0.5 %, respectively. The mAP of R-NMS on Faster-RCNN is 80.83 %, and the effectiveness of the proposed algorithm in the improvement of the mAP of various object detectors is confirmed.
    Shuzhi SU, Runbin CHEN, Yanmin ZHU, Bowen JIANG. Relocation non-maximum suppression algorithm[J]. Optics and Precision Engineering, 2022, 30(13): 1620
    Download Citation