• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2021, Vol. 43 ›› Issue (10): 1803-1809.

Previous Articles     Next Articles

An improved Mask RCNN algorithm based on adaptive-threshold non-maximum suppression

WANG Mei1,LI Dong-xu1,CHEN Lin-lin1,FAN Si-meng1,XU Chuan-hai1,YANG Er-long2#br#

#br#
  

  1. (1.School of Computer and Information Technology,Northeast Petroleum University,Daqing 163318;

    2.School of Petroleum Engineering,Northeast Petroleum University,Daqing 163318,China)
  • Received:2020-08-15 Revised:2020-11-24 Accepted:2021-10-25 Online:2021-10-25 Published:2021-10-22

Abstract: Target detection algorithms under big data often have the problem of missed target detection and repeated detection. To solve this problem, Mask RCNNAT-NMS algorithm based on AT-NMS is proposed. Firstly, a deformable convolution module is added on the basis of ResNet to enhance the ability to extract multi-layer convolution features of the target. Secondly, 
the AT-NMS algorithm is used to extract the in-depth information of the candidate target area in the RPN (Regional Candidate Network) stage. Thirdly, the positioning of the target is more accurate through two quantitative processing of ROI Align. Finally, three branches are used to achieve target instance segmentation, target classification and target border regression. The experimental results on the PASCAL-VOC2012 and Indoor CVPR_09 data sets show that, compared with the mask RCNN algorithm, the Mask RCNNAT-NMS algorithm reduces the repeated detection rate and the target missed detection rate, and improves the recognition accuracy. It can be seen that Mask RCNNAT-NMS algorithm can alleviate the problem of target missing and repeated detection caused by fixed threshold, and improve the detection accuracy on this basis.


Key words: target detection, Mask RCNN, adaptive threshold non-maximum supression, residual network