随着社会技术力量的加速发展,无人机航拍图像逐渐渗透到各个行业。由于无人机的速度可变,捕获的图像被遮蔽,模糊,和模糊。第二,无人机在不同的高度飞行,导致目标尺度变化,难以检测和识别小目标。为了解决上述问题,提出了一种改进的ASG-YOLOv5模型。首先,本研究提出了一个动态的上下文注意模块,利用特征分数动态分配特征权重,通过通道维度输出特征信息,提高模型对小目标特征信息的关注度,增加网络提取上下文信息的能力;其次,本研究设计了空间门控滤波多方向加权融合模块,多尺度融合阶段采用空间滤波和加权双向融合,提高弱目标的表征,减少冗余信息的干扰,更好地适应无人机遥感航拍对图像中微弱目标的检测;同时,使用归一化Wasserstein距离和CIOU回归损失函数,通过对回归框架的高斯分布进行建模,得到回归框架的相似度度量值,增加了对小目标位置差异的平滑,解决了小目标位置偏差非常敏感的问题,从而有效地提高了该模型对小目标的检测精度。本文在VisDrone2021和AI-TOD数据集上对模型进行训练和测试。本研究使用NWPU-RESISC数据集进行视觉检测验证。实验结果表明,ASG-YOLOv5在无人机遥感航拍图像中具有较好的检测效果,帧/秒(FPS)达到86,满足小目标检测的实时性要求,能更好地适应航空影像数据集中弱小目标的检测,ASG-YOLOv5优于许多现有的目标检测方法,检测精度达到21.1%mAP值。mAP值分别提高了2.9%和1.4%,分别,与YOLOV5模型相比。该项目可在https://github.com/woaini-shw/asg-yolov5上获得。git.
With the accelerated development of the technological power of society, aerial images of drones gradually penetrated various industries. Due to the variable speed of drones, the captured images are shadowed, blurred, and obscured. Second, drones fly at varying altitudes, leading to changing target scales and making it difficult to detect and identify small targets. In order to solve the above problems, an improved ASG-YOLOv5 model is proposed in this paper. Firstly, this research proposes a dynamic contextual attention module, which uses feature scores to dynamically assign feature weights and output feature information through channel dimensions to improve the model\'s attention to small target feature information and increase the network\'s ability to extract contextual information; secondly, this research designs a spatial gating filtering multi-directional weighted fusion module, which uses spatial filtering and weighted bidirectional fusion in the multi-scale fusion stage to improve the characterization of weak targets, reduce the interference of redundant information, and better adapt to the detection of weak targets in images under unmanned aerial vehicle remote sensing aerial photography; meanwhile, using Normalized Wasserstein Distance and CIoU regression loss function, the similarity metric value of the regression frame is obtained by modeling the Gaussian distribution of the regression frame, which increases the smoothing of the positional difference of the small targets and solves the problem that the positional deviation of the small targets is very sensitive, so that the model\'s detection accuracy of the small targets is effectively improved. This paper trains and tests the model on the VisDrone2021 and AI-TOD datasets. This study used the NWPU-RESISC dataset for visual detection validation. The experimental results show that ASG-YOLOv5 has a better detection effect in unmanned aerial vehicle remote sensing aerial images, and the frames per second (FPS) reaches 86, which meets the requirement of real-time small target detection, and it can be better adapted to the detection of the weak and small targets in the aerial image dataset, and ASG-YOLOv5 outperforms many existing target detection methods, and its detection accuracy reaches 21.1% mAP value. The mAP values are improved by 2.9% and 1.4%, respectively, compared with the YOLOv5 model. The project is available at https://github.com/woaini-shw/asg-yolov5.git.