Real-Time Object Detection for Millimeter-Wave Images Based on Improved Faster Regions with Convolutional Neural Networks

被引:8
作者
Hou Bingji [1 ,2 ,3 ]
Yang Minghui [1 ]
Sun Xiaowei [1 ]
机构
[1] Chinese Acad Sci, Shanghai Inst Microsyst & Informat Technol SIMIT, Key Lab Terahertz Solid Technol, Shanghai 200050, Peoples R China
[2] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
[3] ShanghaiTech Univ, Sch Informat Sci & Technol, Shanghai 201210, Peoples R China
关键词
imaging processing; image recognition; convolutional neural network; deconvolution; millimeter wave image; object detection;
D O I
10.3788/LOP56.131009
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
An efficient and fast convolution neural network for millimeter-wave images that uses deconvolution and a shortcut connection is proposed. The proposed network retains the low-order fine-grained features of the image and significantly improves the detection speed to 27 frame/s from 9 frame/s of original frame. The RCNN (Regions with Convolutional Neural Networks) part of the Faster RCNN is removed. To achieve better network convergence, the initial candidate box size is designed based on thought clustering. The online hard example mining is applied to optimize the loss function of the Faster RCNN such that the imbalance problem between positive and negative samples in millimeter wave images is solved and the training speed is improved significantly. By using the proposed algorithm, the accuracy of 87. 6% and the detection rate of 81. 2% arc obtained on the test set. Compared with mainstream algorithms, the proposed algorithm improves the F, score by approximately 5%.
引用
收藏
页数:7
相关论文
共 21 条
[1]  
Albelwi S, 2016, 2016 15TH IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS (ICMLA 2016), P53, DOI [10.1109/ICMLA.2016.46, 10.1109/ICMLA.2016.0018]
[2]  
[Anonymous], 2016, CoRR abs/1512.00567, DOI DOI 10.1109/CVPR.2016.308
[3]  
Azulay Aharon., 2018, 180512177 ARXIV
[4]  
He K., 2016, IEEE C COMPUT VIS PA, DOI [10.1007/978-3-319-46493-0_38, DOI 10.1007/978-3-319-46493-0_38, DOI 10.1109/CVPR.2016.90]
[5]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269
[6]  
Jia Y Q, 2011, P 22 ACM INT C MULT, P675
[7]  
Joffe S, 2015, P 32 INT C INT C MAC, V37, P118
[8]  
Joseph RK, 2016, CRIT POL ECON S ASIA, P1
[9]  
LIN TY, 2017, PROC CVPR IEEE, P936, DOI DOI 10.1109/CVPR.2017.106
[10]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37