Lightweight high-precision SAR ship detection method based on YOLOv7-LDS.
The current challenges in Synthetic Aperture Radar (SAR) ship detection tasks revolve around handling significant variations in target sizes and managing high computational expenses, which hinder practical deployment on satellite or mobile airborne platforms. In response to these challenges, this re...
| Main Authors: | , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2024-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0296992&type=printable |
| _version_ | 1797291730724192256 |
|---|---|
| author | Shiliang Zhu Min Miao |
| author_facet | Shiliang Zhu Min Miao |
| author_sort | Shiliang Zhu |
| collection | DOAJ |
| description | The current challenges in Synthetic Aperture Radar (SAR) ship detection tasks revolve around handling significant variations in target sizes and managing high computational expenses, which hinder practical deployment on satellite or mobile airborne platforms. In response to these challenges, this research presents YOLOv7-LDS, a lightweight yet highly accurate SAR ship detection model built upon the YOLOv7 framework. In the core of YOLOv7-LDS's architecture, we introduce a streamlined feature extraction network that strikes a delicate balance between detection precision and computational efficiency. This network is founded on Shufflenetv2 and incorporates Squeeze-and-Excitation (SE) attention mechanisms as its key elements. Additionally, in the Neck section, we introduce the Weighted Efficient Aggregation Network (DCW-ELAN), a fundamental feature extraction module that leverages Coordinate Attention (CA) and Depthwise Convolution (DWConv). This module efficiently aggregates features while preserving the ability to identify small-scale variations, ensuring top-quality feature extraction. Furthermore, we introduce a lightweight Spatial Pyramid Dilated Convolution Cross-Stage Partial Channel (LSPHDCCSPC) module. LSPHDCCSPC is a condensed version of the Spatial Pyramid Pooling Cross-Stage Partial Channel (SPPCSPC) module, incorporating Dilated Convolution (DConv) as a central component for extracting multi-scale information. The experimental results show that YOLOv7-LDS achieves a remarkable Mean Average Precision (mAP) of 99.1% and 95.8% on the SAR Ship Detection Dataset (SSDD) and the NWPU VHR-10 dataset with a parameter count (Params) of 3.4 million, a Giga Floating Point Operations Per Second (GFLOPs) of 6.1 and an Inference Time (IT) of 4.8 milliseconds. YOLOv7-LDS effectively strikes a fine balance between computational cost and detection performance, surpassing many of the current state-of-the-art object detection models. As a result, it offers a more resilient solution for maritime ship monitoring. |
| first_indexed | 2024-03-07T19:41:00Z |
| format | Article |
| id | doaj.art-d07abdefb077434ab16acd16aa5c2e9a |
| institution | Directory Open Access Journal |
| issn | 1932-6203 |
| language | English |
| last_indexed | 2024-03-07T19:41:00Z |
| publishDate | 2024-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj.art-d07abdefb077434ab16acd16aa5c2e9a2024-02-29T05:31:58ZengPublic Library of Science (PLoS)PLoS ONE1932-62032024-01-01192e029699210.1371/journal.pone.0296992Lightweight high-precision SAR ship detection method based on YOLOv7-LDS.Shiliang ZhuMin MiaoThe current challenges in Synthetic Aperture Radar (SAR) ship detection tasks revolve around handling significant variations in target sizes and managing high computational expenses, which hinder practical deployment on satellite or mobile airborne platforms. In response to these challenges, this research presents YOLOv7-LDS, a lightweight yet highly accurate SAR ship detection model built upon the YOLOv7 framework. In the core of YOLOv7-LDS's architecture, we introduce a streamlined feature extraction network that strikes a delicate balance between detection precision and computational efficiency. This network is founded on Shufflenetv2 and incorporates Squeeze-and-Excitation (SE) attention mechanisms as its key elements. Additionally, in the Neck section, we introduce the Weighted Efficient Aggregation Network (DCW-ELAN), a fundamental feature extraction module that leverages Coordinate Attention (CA) and Depthwise Convolution (DWConv). This module efficiently aggregates features while preserving the ability to identify small-scale variations, ensuring top-quality feature extraction. Furthermore, we introduce a lightweight Spatial Pyramid Dilated Convolution Cross-Stage Partial Channel (LSPHDCCSPC) module. LSPHDCCSPC is a condensed version of the Spatial Pyramid Pooling Cross-Stage Partial Channel (SPPCSPC) module, incorporating Dilated Convolution (DConv) as a central component for extracting multi-scale information. The experimental results show that YOLOv7-LDS achieves a remarkable Mean Average Precision (mAP) of 99.1% and 95.8% on the SAR Ship Detection Dataset (SSDD) and the NWPU VHR-10 dataset with a parameter count (Params) of 3.4 million, a Giga Floating Point Operations Per Second (GFLOPs) of 6.1 and an Inference Time (IT) of 4.8 milliseconds. YOLOv7-LDS effectively strikes a fine balance between computational cost and detection performance, surpassing many of the current state-of-the-art object detection models. As a result, it offers a more resilient solution for maritime ship monitoring.https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0296992&type=printable |
| spellingShingle | Shiliang Zhu Min Miao Lightweight high-precision SAR ship detection method based on YOLOv7-LDS. PLoS ONE |
| title | Lightweight high-precision SAR ship detection method based on YOLOv7-LDS. |
| title_full | Lightweight high-precision SAR ship detection method based on YOLOv7-LDS. |
| title_fullStr | Lightweight high-precision SAR ship detection method based on YOLOv7-LDS. |
| title_full_unstemmed | Lightweight high-precision SAR ship detection method based on YOLOv7-LDS. |
| title_short | Lightweight high-precision SAR ship detection method based on YOLOv7-LDS. |
| title_sort | lightweight high precision sar ship detection method based on yolov7 lds |
| url | https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0296992&type=printable |
| work_keys_str_mv | AT shiliangzhu lightweighthighprecisionsarshipdetectionmethodbasedonyolov7lds AT minmiao lightweighthighprecisionsarshipdetectionmethodbasedonyolov7lds |