智能科学与技术学报2024，Vol.6Issue(2)：262-271,10.DOI:10.11959/j.issn.2096-6652.202422
UAVAI-YOLO:无人机航拍图像的小目标检测模型
UAVAI-YOLO:dense small target detection algorithm based on UAV aerial images
摘要
Abstract
An improved UAVAI-YOLO model was proposed to address the problem of poor target detection in UAV aerial images.Firstly,in order to obtain richer semantic information for the model,the original convolution of the C2f module of the original backbone part was replaced with the improved DCN convolution.Secondly,in order to increase the P2 feature layer without increasing the number of model parameters,the Conv_C module was proposed to downscale the output channel of the backbone network,and at the same time,because of avoiding the loss of semantic information due to channel downsizing,the original convolution of the C2f module in the neck part was replaced by the improved ODConv dynamic convolution.Then,the BIFPN module was introduced to make full use of the contextual semantic in-formation.Finally,Wise-IoU was used to replace the original loss function to improve the accuracy of the model target de-tection frame.Experimental results on the publicly available VisDrone2019 dataset and UAVDT dataset showed that the UAVAI-YOLO model improves 4.4%and 1.1%compared to the original YOLOv8n model mAP0.5,respectively,high de-tectability accuracy compared to other mainstream object detection models.关键词
无人机航拍图像/小目标检测/YOLOv8/可变形卷积网络/注意力机制Key words
UAV aerial images/small object detection/YOLOv8/DCN/attention mechanisms分类
信息技术与安全科学引用本文复制引用
何植仟,曹立杰..UAVAI-YOLO:无人机航拍图像的小目标检测模型[J].智能科学与技术学报,2024,6(2):262-271,10.基金项目
辽宁省教育厅科研项目(No.LJKZ0731) Liaoning Province Department of Education Reasearch Fund(No.LJKZ0731) （No.LJKZ0731）
