A lightweight object detection network based on YOLOv5 for SAR image

Aijia Shen, Liangbo Zhao, Fanyun Xu, Guoqing Wang, Wenchao Liu*, Zimeng Shen

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Remote sensing image target detection is one of the key technologies in the field of intelligent interpretation of remote sensing images, and it has significant application value in various areas, including military defense. When performing remote sensing image object detection on airborne and spaceborne platforms, the vast amount of remote sensing data processing and limited computational resources impose high real-time requirements on the object detection algorithms. This paper designs a lightweight object detection network model named YOLOv5-tiny, based on the existing deep learning network detection model YOLOv5s, and deploys it on the Jetson TX2 development board for training and testing. Experimental results show that the proposed YOLOv5-tiny model, when tested on the SAR-AIRcraft-1.0 with an input image size of 640x640, is 10 times smaller than YOLOv5s; it has a computational cost of 5.2 GFLOPs, which is 1/5 of YOLOv5s, and the processing time for a single image is reduced to half that of YOLOv5s, with only a 0.1% decrease in accuracy.

Original languageEnglish
Title of host publicationIEEE International Conference on Signal, Information and Data Processing, ICSIDP 2024
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9798331515669
DOIs
Publication statusPublished - 2024
Event2nd IEEE International Conference on Signal, Information and Data Processing, ICSIDP 2024 - Zhuhai, China
Duration: 22 Nov 202424 Nov 2024

Publication series

NameIEEE International Conference on Signal, Information and Data Processing, ICSIDP 2024

Conference

Conference2nd IEEE International Conference on Signal, Information and Data Processing, ICSIDP 2024
Country/TerritoryChina
CityZhuhai
Period22/11/2424/11/24

Keywords

  • Jetson TX2
  • lightweight
  • object detection
  • SAR image
  • YOLOv5

Fingerprint

Dive into the research topics of 'A lightweight object detection network based on YOLOv5 for SAR image'. Together they form a unique fingerprint.

Cite this

Shen, A., Zhao, L., Xu, F., Wang, G., Liu, W., & Shen, Z. (2024). A lightweight object detection network based on YOLOv5 for SAR image. In IEEE International Conference on Signal, Information and Data Processing, ICSIDP 2024 (IEEE International Conference on Signal, Information and Data Processing, ICSIDP 2024). Institute of Electrical and Electronics Engineers Inc.. https://doi.org/10.1109/ICSIDP62679.2024.10868616