YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds
To improve the detection ability of infrared small targets in complex backgrounds, an improved detection algorithm YOLO-SASE is proposed in this paper. The algorithm is based on the YOLO detection framework and SRGAN network, taking super-resolution reconstructed images as input, combined with the S...
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2022-06-01
|
| Series: | Sensors |
| Subjects: | |
| Online Access: | https://www.mdpi.com/1424-8220/22/12/4600 |
| _version_ | 1797482321138417664 |
|---|---|
| author | Xiao Zhou Lang Jiang Caixia Hu Shuai Lei Tingting Zhang Xingang Mou |
| author_facet | Xiao Zhou Lang Jiang Caixia Hu Shuai Lei Tingting Zhang Xingang Mou |
| author_sort | Xiao Zhou |
| collection | DOAJ |
| description | To improve the detection ability of infrared small targets in complex backgrounds, an improved detection algorithm YOLO-SASE is proposed in this paper. The algorithm is based on the YOLO detection framework and SRGAN network, taking super-resolution reconstructed images as input, combined with the SASE module, SPP module, and multi-level receptive field structure while adjusting the number of detection output layers through exploring feature weight to improve feature utilization efficiency. Compared with the original model, the accuracy and recall rate of the algorithm proposed in this paper were improved by 2% and 3%, respectively, in the experiment, and the stability of the results was significantly improved in the training process. |
| first_indexed | 2024-03-09T22:31:35Z |
| format | Article |
| id | doaj.art-1ec4307cdb06451a804ffc33417f54bf |
| institution | Directory Open Access Journal |
| issn | 1424-8220 |
| language | English |
| last_indexed | 2024-03-09T22:31:35Z |
| publishDate | 2022-06-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Sensors |
| spelling | doaj.art-1ec4307cdb06451a804ffc33417f54bf2023-11-23T18:55:57ZengMDPI AGSensors1424-82202022-06-012212460010.3390/s22124600YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex BackgroundsXiao Zhou0Lang Jiang1Caixia Hu2Shuai Lei3Tingting Zhang4Xingang Mou5School of Mechanical and Electronic Engineering, Wuhan University of Technology, Wuhan 430070, ChinaSchool of Mechanical and Electronic Engineering, Wuhan University of Technology, Wuhan 430070, ChinaBeijing Aerospace Automatic Control Institute, Beijing 100000, ChinaSchool of Mechanical and Electronic Engineering, Wuhan University of Technology, Wuhan 430070, ChinaSchool of Mechanical and Electronic Engineering, Wuhan University of Technology, Wuhan 430070, ChinaSchool of Mechanical and Electronic Engineering, Wuhan University of Technology, Wuhan 430070, ChinaTo improve the detection ability of infrared small targets in complex backgrounds, an improved detection algorithm YOLO-SASE is proposed in this paper. The algorithm is based on the YOLO detection framework and SRGAN network, taking super-resolution reconstructed images as input, combined with the SASE module, SPP module, and multi-level receptive field structure while adjusting the number of detection output layers through exploring feature weight to improve feature utilization efficiency. Compared with the original model, the accuracy and recall rate of the algorithm proposed in this paper were improved by 2% and 3%, respectively, in the experiment, and the stability of the results was significantly improved in the training process.https://www.mdpi.com/1424-8220/22/12/4600infrared small target detectionsuper-resolution reconstructionadaptive channel attention |
| spellingShingle | Xiao Zhou Lang Jiang Caixia Hu Shuai Lei Tingting Zhang Xingang Mou YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds Sensors infrared small target detection super-resolution reconstruction adaptive channel attention |
| title | YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds |
| title_full | YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds |
| title_fullStr | YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds |
| title_full_unstemmed | YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds |
| title_short | YOLO-SASE: An Improved YOLO Algorithm for the Small Targets Detection in Complex Backgrounds |
| title_sort | yolo sase an improved yolo algorithm for the small targets detection in complex backgrounds |
| topic | infrared small target detection super-resolution reconstruction adaptive channel attention |
| url | https://www.mdpi.com/1424-8220/22/12/4600 |
| work_keys_str_mv | AT xiaozhou yolosaseanimprovedyoloalgorithmforthesmalltargetsdetectionincomplexbackgrounds AT langjiang yolosaseanimprovedyoloalgorithmforthesmalltargetsdetectionincomplexbackgrounds AT caixiahu yolosaseanimprovedyoloalgorithmforthesmalltargetsdetectionincomplexbackgrounds AT shuailei yolosaseanimprovedyoloalgorithmforthesmalltargetsdetectionincomplexbackgrounds AT tingtingzhang yolosaseanimprovedyoloalgorithmforthesmalltargetsdetectionincomplexbackgrounds AT xingangmou yolosaseanimprovedyoloalgorithmforthesmalltargetsdetectionincomplexbackgrounds |