基于YOLOv4的空间红外弱目标检测

刘杨帆; 曹立华; 李宁; 张云峰

doi:10.37188/CJLCD.2020-0227

您当前的位置：

首页 >

文章列表页 >

基于YOLOv4的空间红外弱目标检测

图像处理 | 更新时间：2021-04-02

- 基于YOLOv4的空间红外弱目标检测
- Detection of space infrared weak target based on YOLOv4
- 液晶与显示 2021年36卷第4期页码：615-623
- 作者机构：
  
  1.中国科学院长春光学精密机械与物理研究所, 吉林长春 130033
  2.中国科学院大学, 北京 100049
  3.激光与物质相互作用国家重点实验室, 吉林长春 130033
- 作者简介：
  
  [ "刘杨帆(1994-), 男, 内蒙古包头人, 硕士研究生, 2018年于内蒙古大学获得学士学位, 主要从事计算机视觉、目标特性识别、深度学习等方面的研究。E-mail:1370668604@qq.com" ]
  [ "曹立华(1971-), 男, 吉林磐石人, 博士, 研究员, 2014年于长春理工大学获得博士学位, 主要从事光电仪器总体集成技术、目标特性测量与识别等方面的研究。E-mail: cao0983@sina.com" ]
- 基金信息：
  
  国家自然科学基金资助项目(61705219);吉林省科技发展计划项目(20190103133JH)
- DOI：10.37188/CJLCD.2020-0227
  中图分类号： TP391.4
- 收稿日期：2020-09-07，
  
  修回日期：2020-10-29，
  
  纸质出版日期：2021-04
- 稿件说明：
移动端阅览
刘杨帆, 曹立华, 李宁, 等. 基于YOLOv4的空间红外弱目标检测[J]. 液晶与显示, 2021,36(4):615-623.

Yang-fan LIU, Li-hua CAO, Ning LI, et al. Detection of space infrared weak target based on YOLOv4[J]. Chinese journal of liquid crystals and displays, 2021, 36(4): 615-623.
刘杨帆, 曹立华, 李宁, 等. 基于YOLOv4的空间红外弱目标检测[J]. 液晶与显示, 2021,36(4):615-623. DOI： 10.37188/CJLCD.2020-0227.

Yang-fan LIU, Li-hua CAO, Ning LI, et al. Detection of space infrared weak target based on YOLOv4[J]. Chinese journal of liquid crystals and displays, 2021, 36(4): 615-623. DOI： 10.37188/CJLCD.2020-0227.

摘要

在空间红外弱目标检测问题中，相关模板法和帧间差分法等传统算法判别率较低，且对数据质量要求较高。针对这一问题，本文提出了一种基于改进YOLOV4的空间红外弱目标检测方法，该算法首先针对空间不同红外目标建立了相应的数据集；以YOLOv4为基础建立了空间目标检测任务专用的神经网络框架，利用k-means聚类算法重新构造先验框；针对红外弱目标的特性设计了多尺度融合算法来提高弱目标的检测精度；最后应用COCO数据集和实验室采集到的红外图像数据集对本算法进行了训练和测试。试验结果表明，本文改进算法较YOLOv4算法在检测的准确性上有明显提升，其平均准确率（AP）可达93.25%以上，检测速度达到了38.99 ms/frame，验证了算法对于空间红外弱目标检测的有效性，很好地满足了空间红外弱目标检测任务的需求。

Abstract

In the problem of space infrared weak target detection

traditional algorithms such as correlation template method and center of gravity

etc.

have a low discrimination rate and high data quality requirements. To solve this problem

the space infrared weak target detection algorithm based on improved YOLOv4 is proposed in this paper. The algorithm first establishes corresponding data sets for different infrared targets in space. Based on YOLOv4

a special neural network framework for space target detection tasks is established. The k-means clustering algorithm is used to reconstruct the prior frame

and multi-scale fusion is designed according to the characteristics of infrared weak targets to improve the detection accuracy of weak targets. Finally

COCO data set and the infrared image data set collected in the laboratory are used to train the algorithm and test. The test results show that the improved algorithm has a significant improvement in the accuracy of detection compared with the YOLOv4 algorithm. Its average accuracy (AP) can reach more than 93.25%

and the detection speed has reached 38.99 ms/frame

which verifies that the effectiveness of target detection satisfies the needs of space infrared weak target detection tasks.

关键词

Keywords

references

刘炎 , 王涛 , 陈凡胜 , 等 . 基于时域信噪比的红外弱小目标检测 . 太赫兹科学与电子信息学报 , 2019 . 7 ( 1 ): 118 - 123 . https://www.cnki.com.cn/Article/CJFDTOTAL-XXYD201901027.htm https://www.cnki.com.cn/Article/CJFDTOTAL-XXYD201901027.htm .

Y LIU , T WANG , F S CHEN , 等 . Infrared small target detection algorithm based on spatial Signal to Noise Ratio . Journal of Terahertz Science and Electronic Information Technology , 2019 . 17 ( 1 ): 118 - 123 . https://www.cnki.com.cn/Article/CJFDTOTAL-XXYD201901027.htm https://www.cnki.com.cn/Article/CJFDTOTAL-XXYD201901027.htm .

荣生辉 , 刘刚 , 周彗鑫 , 等 . 基于改进的剪切波变换和引导滤波的红外弱小目标背景抑制方法 . 光子学报 , 2015 . 44 ( 2 ): 0210002 https://www.cnki.com.cn/Article/CJFDTOTAL-GZXB201502013.htm https://www.cnki.com.cn/Article/CJFDTOTAL-GZXB201502013.htm .

S H RONG , G LIU , H X ZHOU , 等 . Infrared dim and small target background suppression based on the improved shearlet transform and the guide filter . Acta Photonica Sinica , 2015 . 44 ( 2 ): 0210002 https://www.cnki.com.cn/Article/CJFDTOTAL-GZXB201502013.htm https://www.cnki.com.cn/Article/CJFDTOTAL-GZXB201502013.htm .

曾溢良 , 蓝金辉 , 邹金霖 . 滑动置信度约束的红外弱小目标跟踪算法研究 . 兵工学报 , 2017 . 38 ( 9 ): 1771 - 1778 . DOI: 10.3969/j.issn.1000-1093.2017.09.014 http://doi.org/10.3969/j.issn.1000-1093.2017.09.014 .

Y L ZENG , J H LAN , J L ZOU . Research on infrared dim-small target tracking algorithm with template sliding confidence constraint . Acta Armamentarii , 2017 . 38 ( 9 ): 1771 - 1778 . DOI: 10.3969/j.issn.1000-1093.2017.09.014 http://doi.org/10.3969/j.issn.1000-1093.2017.09.014 .

侯旺 , 孙晓亮 , 尚洋 , 等 . 红外弱小目标检测技术研究现状与发展趋势 . 红外技术 , 2015 . 37 ( 1 ): 1 - 10 . https://www.cnki.com.cn/Article/CJFDTOTAL-HWJS201501001.htm https://www.cnki.com.cn/Article/CJFDTOTAL-HWJS201501001.htm .

W HOU , X L SUN , Y SHANG , 等 . Present state and perspectives of small infrared targets detection technology . Infrared Technology , 2015 . 37 ( 1 ): 1 - 10 . https://www.cnki.com.cn/Article/CJFDTOTAL-HWJS201501001.htm https://www.cnki.com.cn/Article/CJFDTOTAL-HWJS201501001.htm .

孙继刚. 序列图像红外小目标检测与跟踪算法研究[D]. 长春: 中国科学院长春光学精密机械与物理研究所, 2014.

SUN J G. Research on algorithm of infrared small target detection and tracking in image sequence[D]. Changchun: Changchun Institute of Optics, Fine Mechanics and Physics Chinese Academy of Sciences, 2014. (in Chinese)

A L DA CUNHA , J ZHOU , M N DO . The nonsubsampled contourlet transform: theory, design, and applications . IEEE Transactions on Image Processing , 2006 . 15 ( 10 ): 3089 - 3101 . DOI: 10.1109/TIP.2006.877507 http://doi.org/10.1109/TIP.2006.877507 .

Barnett J T. Statistical analysis of median subtraction filtering with application to point target detection in infrared backgrounds[C]// Proceedings of SPIE 1050, Infrared Systems and Components Ⅲ. Los Angeles, CA, United States: SPIE, 1989.

X Z BAI , F G ZHOU . Analysis of new top-hat transformation and the application for infrared dim small target detection . Pattern Recognition , 2010 . 43 ( 6 ): 2145 - 2156 . DOI: 10.1016/j.patcog.2009.12.023 http://doi.org/10.1016/j.patcog.2009.12.023 .

B PORAT , B FRIEDLANDER . A frequency domain algorithm for multiframe detection and estimation of dim targets . IEEE Transactions on Pattern Analysis and Machine Intelligence , 1990 . 12 ( 4 ): 398 - 401 . DOI: 10.1109/34.50625 http://doi.org/10.1109/34.50625 .

BOCHKOVSKIY A, WANG C Y, LIAO H Y M. YOLOv4: optimal speed and accuracy of object detection[EB/OL]. [2020-04-23] . https://github.com/AlexeyAB/darknet https://github.com/AlexeyAB/darknet .

杨州 , 慕晓冬 , 王舒洋 , 等 . 基于多尺度特征融合的遥感图像场景分类 . 光学精密工程 , 2018 . 26 ( 12 ): 3099 - 3107 . https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM201812027.htm https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM201812027.htm .

Z YANG , X D MU , S Y WANG , 等 . Scene classification of remote sensing images based on multiscale features fusion . Optics and Precision Engineering , 2018 . 26 ( 12 ): 3099 - 3107 . https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM201812027.htm https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM201812027.htm .

马立 , 巩笑天 , 欧阳航空 . Tiny YOLOv3目标检测改进 . 光学精密工程 , 2020 . 28 ( 4 ): 988 - 995 . https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM202004027.htm https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM202004027.htm .

L MA , X T GONG , H K OUYANG . Improvement of tiny YOLOv3 target detection . Optics and Precision Engineering , 2020 . 28 ( 4 ): 988 - 995 . https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM202004027.htm https://www.cnki.com.cn/Article/CJFDTOTAL-GXJM202004027.htm .

丁鹏 , 张叶 , 刘让 , 等 . 结合形态学和Canny算法的红外弱小目标检测 . 液晶与显示 , 2016 . 31 ( 8 ): 793 - 800 . http://cjlcd.lightpublishing.cn/thesisDetails#10.3788/YJYXS20163108.0793 http://cjlcd.lightpublishing.cn/thesisDetails#10.3788/YJYXS20163108.0793 .

P DING , Y ZHANG , R LIU , 等 . Infrared small target detection based on adaptive Canny algorithm and morphology . Chinese Journal of Liquid Crystals and Displays , 2016 . 31 ( 8 ): 793 - 800 . http://cjlcd.lightpublishing.cn/thesisDetails#10.3788/YJYXS20163108.0793 http://cjlcd.lightpublishing.cn/thesisDetails#10.3788/YJYXS20163108.0793 .

浏览量

259

下载量

CSCD

文章被引用时，请邮件提醒。

提交

工具集

关联资源

基于Mamba-2的视频快照压缩成像重构方法

无人机绝对视觉定位的研究进展

基于自适应遮挡恢复与拓扑姿态双向感知的三维手部重建方法

基于光流预测的三维光场显示透视关系校正