Optics and Precision Engineering, Volume. 30, Issue 13, 1620(2022)

Relocation non-maximum suppression algorithm

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    Shuzhi SU, Runbin CHEN, Yanmin ZHU, Bowen JIANG. Relocation non-maximum suppression algorithm[J]. Optics and Precision Engineering, 2022, 30(13): 1620

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: Dec. 24, 2021

    Accepted: --

    Published Online: Jul. 27, 2022

    The Author Email: SU Shuzhi (sushuzhi@foxmail.com)

    DOI:10.37188/OPE.20223013.1620

    Topics