• Laser & Optoelectronics Progress
  • Vol. 58, Issue 24, 2420001 (2021)
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
    DOI: 10.3788/LOP202158.2420001 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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