基于改进YOLOv5s的无人机图像实时目标检测

• 作者: 陈旭 彭冬亮 谷雨
• 来源期刊: 光电工程
• 学科: 工学
• 关键词: 无人机图像 实时目标检测 YOLOv5sm+
• 年，卷（期）: 2022,（3）
• 所属期刊栏目: 科研论文|Article
• 页码范围: 67-79
• 页数: 13页
• 分类号: TP391.41
• 语种: 中文
• DOI: 10.12086/oee.2022.210372

摘要

针对无人机图像背景复杂、分辨率高、目标尺度差异大等特点,提出了一种实时目标检测算法YOLOv5sm+.首先,分析了网络宽度和深度对无人机图像检测性能的影响,通过引入可增大感受野的残差空洞卷积模块来提高空间特征的利用率,基于YOLOv5s设计了一种改进的浅层网络YOLOv5sm,以提高无人机图像的检测精度.然后,设计了一种特征融合模块SCAM,通过局部特征自监督的方式提高细节信息利用率,通过多尺度特征有效融合提高了中大目标的分类精度.最后,设计了目标位置回归与分类解耦的检测头结构,进一步提高了分类精度.采用VisDrone无人机航拍数据集实验结果表明,提出的YOLOv5sm+模型对验证集测试时交并比为0.5时的平均精度均值(mAP50)达到了60.6％,相比于YOLOv5s模型mAP50提高了4.8％,超过YOLOv5m模型的精度,同时推理速度也有提升.通过在DIOR遥感数据集上的迁移实验也验证了改进模型的有效性.提出的改进模型具有虚警率低、重叠目标识别率高的特点,适合于无人机图像的目标检测任务.