Lightweight satellite recognition algorithm based on improved YOLOv5

Haitao Liu; Lifen Wang; Zeng Gao; Xiuqian Li; Yanxia Yang; Jun Li

doi:10.1117/12.3050441

24 October 2024 Lightweight satellite recognition algorithm based on improved YOLOv5

Haitao Liu, Lifen Wang, Zeng Gao, Xiuqian Li, Yanxia Yang, Jun Li

Proceedings Volume 13396, Third International Conference on Image Processing, Object Detection, and Tracking (IPODT 2024); 133960K (2024) https://doi.org/10.1117/12.3050441
Event: 3rd International Conference on Image Processing, Object Detection and Tracking (IPODT24), 2024, Nanjing, China

Abstract

Target detection technology plays a crucial role in the aerospace field, and recognizing satellites with few samples is of great significance for ensuring national space security. However, the limited computing resources of satellites restrict the application of intelligent algorithms. To overcome this issue, this paper proposes a lightweight improvement of the YOLOv5s detection algorithm with the introduction of an attention mechanism. First, the lightweight module GhostNet is used to modify the backbone network, serving as the primary feature extraction network of the model. The Coordinate Attention EMA module is introduced in the neck network to enhance the feature representation capability for detecting target satellites, thereby improving detection accuracy in complex backgrounds. Simulations based on actual sampled data show that the algorithm's parameter count is reduced by 27.6%, achieving a mAP@0.5:0.95% of 91.21% and a detection rate of 400 FPS. This effectively enhances the model's multi-scale feature representation and fusion capabilities, improving target detection accuracy.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Haitao Liu, Lifen Wang, Zeng Gao, Xiuqian Li, Yanxia Yang, and Jun Li "Lightweight satellite recognition algorithm based on improved YOLOv5", Proc. SPIE 13396, Third International Conference on Image Processing, Object Detection, and Tracking (IPODT 2024), 133960K (24 October 2024); https://doi.org/10.1117/12.3050441

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available