EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s
Due to the diversity of aircraft target scale and interference of background strong scattering in synthetic aperture radar (SAR) images, it is a challenge for target detection tasks. In response to these problems, this paper proposes a new SAR image aircraft target detection model named EST-YOLOv5s....
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2023-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10274975/ |
| _version_ | 1797655637192081408 |
|---|---|
| author | Min Huang Weihao Yan Wenhui Dai Jingyang Wang |
| author_facet | Min Huang Weihao Yan Wenhui Dai Jingyang Wang |
| author_sort | Min Huang |
| collection | DOAJ |
| description | Due to the diversity of aircraft target scale and interference of background strong scattering in synthetic aperture radar (SAR) images, it is a challenge for target detection tasks. In response to these problems, this paper proposes a new SAR image aircraft target detection model named EST-YOLOv5s. The proposed model integrates the Efficient Channel Attention (ECA) mechanism into the C3 module of the backbone network, which enhances the scattering features of aircraft targets and suppresses irrelevant background information without increasing the number of parameters. Secondly, replace the bottleneck module in the last C3 module in the backbone network with the Swin Transformer Block. By using the shifted window partitioning approach to obtain the global perception ability, the problem of missed detection of small objects is improved. Finally, the Task-Specific Context Decoupling (TSCODE) head is used to balance the relationship between classification and regression so that different contextual details can be better utilized. In this paper, the SAR Aircraft Detection Dataset (SADD) is used as the experimental data set to compare with the baseline model YOLOv5s. The experimental outcomes indicate that the recall of the EST-YOLOv5s model reached 94.2%, the precision reached 97.3%, and the mAP@50 reached 97.8%, which were 2.3%, 1.7%, and 1.7% higher than YOLOv5s respectively. Furthermore, our model also meets the real-time requirements in terms of speed and exhibits strong anti-interference ability. |
| first_indexed | 2024-03-11T17:17:22Z |
| format | Article |
| id | doaj.art-4ee8d53f7a254114985d175424c5f4bc |
| institution | Directory Open Access Journal |
| issn | 2169-3536 |
| language | English |
| last_indexed | 2024-03-11T17:17:22Z |
| publishDate | 2023-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj.art-4ee8d53f7a254114985d175424c5f4bc2023-10-19T23:01:32ZengIEEEIEEE Access2169-35362023-01-011111302711304110.1109/ACCESS.2023.332357510274975EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5sMin Huang0https://orcid.org/0000-0001-6152-7912Weihao Yan1https://orcid.org/0009-0007-4618-4326Wenhui Dai2https://orcid.org/0009-0006-7829-2105Jingyang Wang3https://orcid.org/0000-0003-3829-6540Hebei University of Science and Technology, Shijiazhuang, ChinaHebei University of Science and Technology, Shijiazhuang, ChinaHebei University of Science and Technology, Shijiazhuang, ChinaHebei University of Science and Technology, Shijiazhuang, ChinaDue to the diversity of aircraft target scale and interference of background strong scattering in synthetic aperture radar (SAR) images, it is a challenge for target detection tasks. In response to these problems, this paper proposes a new SAR image aircraft target detection model named EST-YOLOv5s. The proposed model integrates the Efficient Channel Attention (ECA) mechanism into the C3 module of the backbone network, which enhances the scattering features of aircraft targets and suppresses irrelevant background information without increasing the number of parameters. Secondly, replace the bottleneck module in the last C3 module in the backbone network with the Swin Transformer Block. By using the shifted window partitioning approach to obtain the global perception ability, the problem of missed detection of small objects is improved. Finally, the Task-Specific Context Decoupling (TSCODE) head is used to balance the relationship between classification and regression so that different contextual details can be better utilized. In this paper, the SAR Aircraft Detection Dataset (SADD) is used as the experimental data set to compare with the baseline model YOLOv5s. The experimental outcomes indicate that the recall of the EST-YOLOv5s model reached 94.2%, the precision reached 97.3%, and the mAP@50 reached 97.8%, which were 2.3%, 1.7%, and 1.7% higher than YOLOv5s respectively. Furthermore, our model also meets the real-time requirements in terms of speed and exhibits strong anti-interference ability.https://ieeexplore.ieee.org/document/10274975/Aircraft target detectionanti-interferenceSAREST-YOLOv5sECAswin transformer |
| spellingShingle | Min Huang Weihao Yan Wenhui Dai Jingyang Wang EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s IEEE Access Aircraft target detection anti-interference SAR EST-YOLOv5s ECA swin transformer |
| title | EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s |
| title_full | EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s |
| title_fullStr | EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s |
| title_full_unstemmed | EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s |
| title_short | EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s |
| title_sort | est yolov5s sar image aircraft target detection model based on improved yolov5s |
| topic | Aircraft target detection anti-interference SAR EST-YOLOv5s ECA swin transformer |
| url | https://ieeexplore.ieee.org/document/10274975/ |
| work_keys_str_mv | AT minhuang estyolov5ssarimageaircrafttargetdetectionmodelbasedonimprovedyolov5s AT weihaoyan estyolov5ssarimageaircrafttargetdetectionmodelbasedonimprovedyolov5s AT wenhuidai estyolov5ssarimageaircrafttargetdetectionmodelbasedonimprovedyolov5s AT jingyangwang estyolov5ssarimageaircrafttargetdetectionmodelbasedonimprovedyolov5s |