Electronics Optics & Control, Volume. 29, Issue 6, 37(2022)
Remote Sensing Aircraft Detection Algorithm Based on Structural Pruning
Regarding the problem that lightweight algorithm in remote sensing aircraft target detection is difficult to balance accuracy and real-time performancea 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 detectionK-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 additionin order to reduce the parameters of the modelthe scaling factor γ in the normalization layer is used for L1 sparse regularizationthe filter and convolution kernel weights are re-evaluatedchannels with less feature information are iteratively prunedand then the pruning model is fine-tuned to recover accuracy.The experimental results show that after pruningthe model parameters are compressed by 93.1%and the detection speed is 2.46 times faster than that of the original modelwhich can effectively improve the detection accuracy and real-time performance.
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
Category:
Received: May. 18, 2021
Accepted: --
Published Online: Aug. 1, 2022
The Author Email: