Infrared and Laser Engineering, Volume. 50, Issue 8, 20200510(2021)

Infrared object detection network compression using Lp normalized weight

Weipeng Li, Xiaogang Yang, Chuanxiang Li, Ruitao Lu, Xueli Xie, and Chuan He
Author Affiliations
  • Institute of Missile Engineering, Rocket Force Engineering University, Xi’an 710025, China
  • show less

    In view of the characteristic that the infrared image has less texture compared with RGB image, an infrared object detection network compression method using Lp normalized weight was proposed. It aimed at improving the adaptability of convolutional neural network based object detection framework to the infrared images, and compressing the scale of network while improving its generalization ability. Firstly, the phenomenon that the sparsity of Lp normalized weight can be precisely controlled by adjusting p was revealed. Based on the phenomenon, a sparsification method for object detection network was proposed. It respectively trained the backbone network and the detector with Lp spherical gradient descent and classical gradient descent, to balance the network scale and fitting accuracy. The tests on simulated infrared image dataset show that, the proposed method is superior to the dense model on both of network scale and detection accuracy: in terms of network scale, the sparsification reduces the effective parameters of Faster R-CNN, Single Shot multibox Detector (SSD) and YOLOv3 by 52%, 78% and 66% respectively; it also improves the mean Average Precision (mAP) of Faster R-CNN, SSD and YOLOv3 by 0.1%, 0.3% and 0.2%, thus verifying the effectiveness of the proposed method.

    Tools

    Get Citation

    Copy Citation Text

    Weipeng Li, Xiaogang Yang, Chuanxiang Li, Ruitao Lu, Xueli Xie, Chuan He. Infrared object detection network compression using Lp normalized weight[J]. Infrared and Laser Engineering, 2021, 50(8): 20200510

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Infrared technology and application

    Received: Apr. 10, 2021

    Accepted: --

    Published Online: Nov. 2, 2021

    The Author Email:

    DOI:10.3788/IRLA20200510

    Topics