Vehicle-Damage-Detection Segmentation Algorithm Based on Improved Mask RCNN

被引:71
作者
Zhang, Qinghui [1 ,2 ]
Chang, Xianing [2 ]
Bian, Shanfeng Bian [2 ]
机构
[1] Henan Univ Technol, Key Lab Grain Informat Proc & Control, Minist Educ, Zhengzhou 450001, Peoples R China
[2] Henan Univ Technol, Coll Informat Sci & Engn, Zhengzhou 450001, Peoples R China
来源
IEEE ACCESS | 2020年 / 8卷
基金
中国国家自然科学基金;
关键词
Mask RCNN; vehicle-damage-detection; loss function; detection accuracy;
D O I
10.1109/ACCESS.2020.2964055
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Traffic congestion due to vehicular accidents seriously affects normal travel, and accurate and effective mitigating measures and methods must be studied. To resolve traffic accident compensation problems quickly, a vehicle-damage-detection segmentation algorithm based on transfer learning and an improved mask regional convolutional neural network (Mask RCNN) is proposed in this paper. The experiment first collects car damage pictures for preprocessing and uses Labelme to make data set labels, which are divided into training sets and test sets. The residual network (ResNet) is optimized, and feature extraction is performed in combination with Feature Pyramid Network (FPN). Then, the proportion and threshold of the Anchor in the region proposal network (RPN) are adjusted. The spatial information of the feature map is preserved by bilinear interpolation in ROIAlign, and different weights are introduced in the loss function for different-scale targets. Finally, the results of self-made dedicated dataset training and testing show that the improved Mask RCNN has better Average Precision (AP) value, detection accuracy and masking accuracy, and improves the efficiency of solving traffic accident compensation problems.
引用
收藏
页码:6997 / 7004
页数:8
相关论文
共 27 条
  • [1] Water quality monitoring method based on feedback self correcting dense connected convolution network
    Cheng Shuhong
    Zhang Shijun
    Zhang Dianfan
    [J]. NEUROCOMPUTING, 2019, 349 : 301 - 313
  • [2] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [3] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [4] A method based on Multi-Convolution layers Joint and Generative Adversarial Networks for Vehicle Detection
    Han, Guang
    Su, Jinpeng
    Zhang, Chengwei
    [J]. KSII TRANSACTIONS ON INTERNET AND INFORMATION SYSTEMS, 2019, 13 (04): : 1795 - 1811
  • [5] He KM, 2017, IEEE I CONF COMP VIS, P2980, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
  • [6] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [7] Identifying pneumonia in chest X-rays: A deep learning approach
    Jaiswal, Amit Kumar
    Tiwari, Prayag
    Kumar, Sachin
    Gupta, Deepak
    Khanna, Ashish
    Rodrigues, Joel J. P. C.
    [J]. MEASUREMENT, 2019, 145 : 511 - 518
  • [8] A Multi-Organ Nucleus Segmentation Challenge
    Kumar, Neeraj
    Verma, Ruchika
    Anand, Deepak
    Zhou, Yanning
    Onder, Omer Fahri
    Tsougenis, Efstratios
    Chen, Hao
    Heng, Pheng-Ann
    Li, Jiahui
    Hu, Zhiqiang
    Wang, Yunzhi
    Koohbanani, Navid Alemi
    Jahanifar, Mostafa
    Tajeddin, Neda Zamani
    Gooya, Ali
    Rajpoot, Nasir
    Ren, Xuhua
    Zhou, Sihang
    Wang, Qian
    Shen, Dinggang
    Yang, Cheng-Kun
    Weng, Chi-Hung
    Yu, Wei-Hsiang
    Yeh, Chao-Yuan
    Yang, Shuang
    Xu, Shuoyu
    Yeung, Pak Hei
    Sun, Peng
    Mahbod, Amirreza
    Schaefer, Gerald
    Ellinger, Isabella
    Ecker, Rupert
    Smedby, Orjan
    Wang, Chunliang
    Chidester, Benjamin
    That-Vinh Ton
    Minh-Triet Tran
    Ma, Jian
    Minh N Do
    Graham, Simon
    Quoc Dang Vu
    Kwak, Jin Tae
    Gunda, Akshaykumar
    Chunduri, Raviteja
    Hu, Corey
    Zhou, Xiaoyang
    Lotfi, Dariush
    Safdari, Reza
    Kascenas, Antanas
    O'Neil, Alison
    [J]. IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (05) : 1380 - 1391
  • [9] Li, 2019, P SCI SURV MAPP APR, P1
  • [10] Fully Convolutional Instance-aware Semantic Segmentation
    Li, Yi
    Qi, Haozhi
    Dai, Jifeng
    Ji, Xiangyang
    Wei, Yichen
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 4438 - 4446