Anchor-free lightweight infrared object detection method (Invited)

被引：0
|
作者：
Gao F. [1 ,2 ]
Yang X. [2 ]
Lu R. [2 ]
Wang S. [2 ]
Gao J. [2 ]
Xia H. [2 ]
机构：
[1] Beijing Huahang Radio Measurement Institute, Beijing
[2] Missile Engineering Institute, Rocket Force University of Engineering, Xi'an
来源：
Hongwai yu Jiguang Gongcheng/Infrared and Laser Engineering | 2022年 / 51卷 / 04期
关键词：
Asymmetric convolution; Infrared target; Lightweight; Neural network; Object detection;
D O I：
10.3788/IRLA20220193
中图分类号：
学科分类号：
摘要：
According to the characteristics of infrared targets, an anchor-free lightweight infrared target detection method was proposed, which improved the detection ability of embedded platform. For the platform with limited computing resources, a new lightweight convolution structure was proposed. Asymmetric convolution was introduced to enhance the feature expression ability of standard convolution, reducing the amount of parameters and computation effectively. A lightweight feature extraction unit was constructed by designing parallel multi-feature path, which generated rich features through channel concatation, then combining with attention module and channel shuffle. SkipBranch was added to promote the transmission of shallow information to the high level and further enrich the characteristics of the high level. Experiments on FLIR dataset showed that the accuracy of the designed lightweight network structure was 81.7%, which exceeded YOLOv4-tiny. However, the model parameters and calculation amount were reduced by 75.0% and 71.1% respectively, and the reasoning time was compressed by 91.3%, which could meet the real-time detection requirements of infrared object on embedded platform. Copyright ©2022 Infrared and Laser Engineering. All rights reserved.
引用
收藏
相关论文
共 22 条
  • [11] Ren S, He K, Girshick R, Et al., Faster R-CNN: Towards real-time object detection with region proposal networks, IEEE Transactions on Pattern Analysis & Machine Intelligence, 39, 6, pp. 1137-1149, (2017)
  • [12] Wang Chen, Zhang Xiufeng, Liu Chao, Et al., Detection method of wheel hub weld defects based on the improved YOLOv3, Optics and Precision Engineering, 29, 8, pp. 1942-1954, (2021)
  • [13] Cheng Yan, Yu Xuelian, Qian Weixian, Et al., Ship wake extraction and detection from infrared remote sensing images, Infrared and Laser Engineering, 51, 2, (2022)
  • [14] Wang Chunzhe, An Junshe, Jiang Xiujie, Et al., Region proposal optimization algorithm based on convolutional neural networks, Chinese Optics, 12, 6, pp. 1348-1361, (2019)
  • [15] Szegedy C, Ioffe S, Vanhoucke V, Et al., Inception-v4, inception-ResNet and the impact of residual connections on learning, AAAI Conference on Artificial Intelligence, pp. 4278-4284, (2017)
  • [16] Zhang Ruiyan, Jiang Xiujie, An Junshe, Et al., Design of global-contextual detection model for optical remote sensing targets, Chinese Optics, 13, 6, pp. 1302-1313, (2020)
  • [17] Li Weipeng, Yang Xiaogang, Li Chuanxiang, Et al., Infrared object detection network compression using Lp normalized weight, Infrared and Laser Engineering, 50, 8, (2021)
  • [18] Yang Lingxiao, Zhang Ru-Yuan, Li Lida, Et al., SimAM: A simple, parameter-free attention module for convolutional neural networks, International Conference on Machine Learning, 139, pp. 11863-11874, (2021)
  • [19] Ju Moran, Luo Haibo, Liu Guangqi, Et al., Infrared dim and small target detection network based on spatial attention mechanism, Optics and Precision Engineering, 29, 4, pp. 843-853, (2021)
  • [20] Lin T Y, Dollar P, Girshick R, Et al., Feature pyramid networks for object detection, IEEE Computer Society Conference on Computer Vision & Pattern Recognition, pp. 936-944, (2017)
123