Electronics Optics & Control, Volume. 29, Issue 6, 37(2022)

Remote Sensing Aircraft Detection Algorithm Based on Structural Pruning

WANG Chenglong, ZHAO Qian, ZHAO Yan, and GUO Tong
Author Affiliations
  • [in Chinese]
  • show less

    Regarding the problem that lightweight algorithm in remote sensing aircraft target detection is difficult to balance accuracy and real-time performancea model compression method based on YOLOv4 structured pruning is presented.In order to make the anchor frame parameters more suitable for remote sensing datasets and take advantage of network multi-scale detectionK-means++ algorithm is used to cluster the datasets and scale adaptive adjustment is designed to restrain the redundancy of the anchor frame caused by too many small targets and close target sizes.In additionin order to reduce the parameters of the modelthe scaling factor γ in the normalization layer is used for L1 sparse regularizationthe filter and convolution kernel weights are re-evaluatedchannels with less feature information are iteratively prunedand then the pruning model is fine-tuned to recover accuracy.The experimental results show that after pruningthe model parameters are compressed by 93.1%and the detection speed is 2.46 times faster than that of the original modelwhich can effectively improve the detection accuracy and real-time performance.

    Tools

    Get Citation

    Copy Citation Text

    WANG Chenglong, ZHAO Qian, ZHAO Yan, GUO Tong. Remote Sensing Aircraft Detection Algorithm Based on Structural Pruning[J]. Electronics Optics & Control, 2022, 29(6): 37

    Download Citation

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

    Category:

    Received: May. 18, 2021

    Accepted: --

    Published Online: Aug. 1, 2022

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2022.06.008

    Topics