| Autor: |
Chunhui Cui, Feiyang Lv, Rugang Wang, Yuanyuan Wang, Feng Zhou, Xuesheng Bian |
| Jazyk: |
angličtina |
| Rok vydání: |
2024 |
| Předmět: |
|
| Zdroj: |
IEEE Access, Vol 12, Pp 140809-140822 (2024) |
| Druh dokumentu: |
article |
| ISSN: |
2169-3536 |
| DOI: |
10.1109/ACCESS.2024.3461337 |
| Popis: |
Aiming at the YOLO (You Only Look Once) algorithm’s low detection accuracy caused by the complex background environment and large target scale difference in optical remote sensing image detection, the lightweight convolution fusion attention mechanism based AMH-YOLOv8 (Attention Mechanisms Hybrid- YOLOv8) target detection algorithm is proposed in this paper. In this algorithm, the BiFormer attention mechanism is added to enhance the detection performance for small targets. Effectively captures local and global information in remote sensing images, improving the accuracy and generalisation of target detection; secondly, using the lightweight convolution GSConv instead of the original normal convolution. Effectively reduces model size while ensuring that model performance is not compromised. And optimised computation and parameter count improvements due to BiFormer; finally, the SPPCSPC space pyramid structure was added. Effectively enhances the feature extraction capability of the model and reduces the probability of missed and false detections.In order to verify the effectiveness of the algorithm, experimental analyses were conducted on two publicly available datasets, namely DIOR and DOTA. Experimental results indicate that the AMH-YOLOv8 algorithm achieved an impressive detection accuracy of 87.6% and 72.9% in mAP@0.5, The results show that the algorithm in this paper has improved the effectiveness of target detection in optical remote sensing images. And it can better cope with the problems of complex background environment, dense distribution of small targets and large differences in target scales. |
| Databáze: |
Directory of Open Access Journals |
| Externí odkaz: |
|
