计算机科学
人工智能
特征(语言学)
像素
比例(比率)
计算机视觉
干扰(通信)
模式识别(心理学)
目标检测
遥感
频道(广播)
电信
哲学
语言学
物理
量子力学
地质学
作者
Hongbin Han,Fuzhen Zhu,Bing Zhu,Hong Wu
出处
期刊:IEEE Geoscience and Remote Sensing Letters
[Institute of Electrical and Electronics Engineers]
日期:2023-01-01
卷期号:20: 1-5
标识
DOI:10.1109/lgrs.2023.3327878
摘要
To solve the problems of low target detection accuracy caused by complex background of remote sensing images, dense distribution of small targets, various target scales and easily affected by environmental factors, this letter proposes a new target detection algorithm based on an improved YOLOv5. First, the coordinate attention mechanism is added in the last layer of the YOLOv5 backbone network. The coordinate attention can effectively capture the spatial relationships between pixels, thus enhancing the location-awareness of the model to suppress the interference of redundant information. Second, the small target detection layer is added to the neck in the original structure, and three scale feature detection heads are increased to four, and the new detection heads are detected on a 160×160 feature map, which reduces the receptive field and can enhance the multi-scale target detection performance of the algorithm to solve the phenomenon that small targets are closely distributed and easy to miss detection. Finally, experiments and validation were conducted in the DOTA dataset. Experiment results show that our improved algorithm can effectively improve the accuracy of remote sensing images target detection. The mean Average Precision (mAP) of the improved YOLOv5 algorithm reached 70.2%, which is 2.8% higher than the original YOLOv5.
科研通智能强力驱动
Strongly Powered by AbleSci AI