激光与光电子学进展, 2019, 56 (13): 131009, 网络出版: 2019-07-11   

基于改进Faster RCNN的毫米波图像实时目标检测 下载: 989次

Real-Time Object Detection for Millimeter-Wave Images Based on Improved Faster Regions with Convolutional Neural Networks
作者单位
1 中国科学院上海微系统与信息技术研究所太赫兹固态技术重点实验室, 上海 200050
2 中国科学院大学, 北京 100049
3 上海科技大学信息科学与技术学院, 上海 201210
摘要
采用反卷积与捷径连接,针对毫米波图像提出了一种高效、快速的卷积神经网络,在保留图像低阶细粒度特征的同时,检测速度由原框架的9 frame/s大幅提升至27 frame/s,并取消了Faster RCNN (Regions with Convolutional Neural Networks)中的RCNN部分。为了使网络更好地收敛,基于聚类思想设计了初始候选框的大小。使用在线困难样本挖掘(OHEM)优化了Faster RCNN的损失函数,解决了毫米波图像中正负样本失衡的问题,大幅提升了训练速度。所提算法在测试集上取得了87.6%的准确率和81.2%的检出率,F1分数相较于主流算法提升了5%左右。
Abstract
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% are obtained on the test set. Compared with mainstream algorithms, the proposed algorithm improves the F1 score by approximately 5%.

侯冰基, 杨明辉, 孙晓玮. 基于改进Faster RCNN的毫米波图像实时目标检测[J]. 激光与光电子学进展, 2019, 56(13): 131009. Bingji Hou, Minghui Yang, Xiaowei Sun. Real-Time Object Detection for Millimeter-Wave Images Based on Improved Faster Regions with Convolutional Neural Networks[J]. Laser & Optoelectronics Progress, 2019, 56(13): 131009.

本文已被 6 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!