Laser & Optoelectronics Progress, Volume. 58, Issue 24, 2420001(2021)

Improved Faster R-CNN Target Detection Algorithm Based on Attention Mechanism and Soft-NMS

Fengsui Wang1,2,3、*, Qisheng Wang1,2,3, Jingang Chen1,2,3, and Furong Liu1,2,3
Author Affiliations
  • 1School of Electrical Engineering, Anhui Polytechnic University, Wuhu, Anhui 241000, China
  • 2Anhui Key Laboratory of Detection Technology and Energy Saving Devices, Wuhu, Anhui 241000, China
  • 3Key Laboratory of Advanced Perception and Intelligent Control of High-end Equipment, Ministry of Education, Wuhu, Anhui 241000, China
  • show less

    Aiming at the problems of missing detection, false detection, and low detection accuracy of the Faster R-CNN target detection network, a soft non-maximum suppression (Soft-NMS) fusion attention mechanism and the Faster R-CNN (Faster Region-Convolutional Neural Network) target detection algorithm is proposed. In order to enhance the global important feature extraction and weaken the irrelevant feature in the feature map by the Faster R-CNN target detection algorithm, an attention mechanism is firstly introduced into the network. Second, aiming at the problem of local information loss caused by the bottleneck structure formed by two fully connected layers in the attention mechanism, a non-dimensional-reduction channel attention and spatial attention series module that can be trained end-to-end with the convolutional neural network is constructed. Then, a Soft-NMS is introduced to replace the traditional non-maximal suppression (NMS) algorithm after the regional suggestion network, which can reduce the target missing detection and improve the location accuracy. Finally, the error detection rate is introduced into the evaluation criteria to further verify the performance of the model. Experimental results show that the Faster R-CNN algorithm based on ResNet-50 can effectively reduce the missed detection and false detection and improve the location accuracy, and the average detection accuracy is significantly improved.

    Tools

    Get Citation

    Copy Citation Text

    Fengsui Wang, Qisheng Wang, Jingang Chen, Furong Liu. Improved Faster R-CNN Target Detection Algorithm Based on Attention Mechanism and Soft-NMS[J]. Laser & Optoelectronics Progress, 2021, 58(24): 2420001

    Download Citation

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

    Category: Optics in Computing

    Received: Jan. 5, 2021

    Accepted: Mar. 3, 2021

    Published Online: Dec. 3, 2021

    The Author Email: Wang Fengsui (fswang@ahpu.edu.cn)

    DOI:10.3788/LOP202158.2420001

    Topics