计算机视觉
计算机科学
人工智能
特征(语言学)
目标检测
卷积(计算机科学)
核(代数)
Viola–Jones对象检测框架
航空影像
棱锥(几何)
背景(考古学)
图像处理
图像分割
对象(语法)
接头(建筑物)
视频跟踪
特征提取
特征检测(计算机视觉)
上下文图像分类
特征学习
对象类检测
变更检测
视觉对象识别的认知神经科学
模式识别(心理学)
人脸检测
图像分辨率
一般化
杂乱
作者
Zetao Kang,Zhihong Dong,Peng Cao,Mengxin Pang
标识
DOI:10.1117/1.jei.34.6.063043
摘要
Small object detection in drone images faces challenges such as complex backgrounds, limited feature extraction, and occlusion interference, which significantly impact the performance of existing object detection algorithms. To address these challenges, we propose a small object detection enhanced algorithm based on you only look once (YOLO), called YOLO-SODE, which aims to improve the detection accuracy of small objects. First, an adaptive kernel convolution fused with spatial context awareness is proposed to enhance feature extraction capabilities. Next, a small object enhancement pyramid structure is designed, combining space-to-depth convolution and global-local joint feature extraction. By introducing the P2 layer feature map output, it improves global feature extraction and multiscale feature fusion. Finally, an occlusion-aware attention mechanism is incorporated to improve the accuracy of feature detection in target regions. Experimental results show that YOLO-SODE outperforms the baseline YOLOv11n by 7.6% and 6.1% in terms of mAP50 and mAP50-95, with the parameter count remaining at a relatively low level, on the VisDrone-DET2019 dataset. In addition, generalization experiments on multiple representative datasets, including tiny object detection in aerial images (AI-TOD), indicate consistent gains of about 5–6 mAP50 points over the baseline, further validating the effectiveness of the proposed method. The proposed algorithm provides an effective and adaptable solution for small object detection in unmanned aerial vehicle aerial images.
科研通智能强力驱动
Strongly Powered by AbleSci AI