基于改进YOLOv7的无人机航拍图像目标检测

被引:8
作者
吴旭红
赵清华
机构
[1] 太原理工大学
关键词
YOLOv7; 无人机; 航拍图像; 小目标检测; SimAM注意力机制;
D O I
暂无
中图分类号
V279 [无人驾驶飞机]; TP183 [人工神经网络与计算]; TP391.41 [];
学科分类号
1111 ; 080203 ;
摘要
针对无人机捕获场景下目标尺度变化剧烈、小目标检测精度低、漏检率高等问题,提出了一种改进的YOLOv7目标检测算法。首先,在原YOLOv7基础上增加一个极小目标检测层,以适应不同尺度下的物体目标,降低小目标的漏检率;其次,在特征融合网络中引入无参注意力机制,并基于该注意力机制构建了一个MP-SimAM模块,使网络融合更多重要的特征信息;最后,提出了一种新的边框回归损失函数SCIoU Loss,进一步提升模型的收敛速度与检测精度。实验结果表明,该模型在VisDrone 2019数据集上表现出色,所提算法模型在测试集上mAP50达44.0%,相比于基准模型YOLOv7提升了2.6个百分点,对于小目标的检测效果提升明显。
引用
收藏
页码:35 / 40+111 +111
页数:7
相关论文
共 10 条
  • [1] 聚焦小目标的航拍图像目标检测算法
    张智
    易华挥
    郑锦
    [J]. 电子学报, 2023, 51 (04) : 944 - 955
  • [2] 改进YOLOv5的无人机影像小目标检测算法
    谢椿辉
    吴金明
    徐怀宇
    [J]. 计算机工程与应用, 2023, (09) : 198 - 206
  • [3] 改进的YOLOv4-tiny算法及其在无人机目标检测中的应用
    杨锐
    黄山
    [J]. 电光与控制, 2022, 29 (12) : 71 - 77
  • [4] 基于深度学习的无人机航拍目标检测研究综述
    江波
    屈若锟
    李彦冬
    李诚龙
    [J]. 航空学报, 2021, 42 (04) : 137 - 151
  • [5] Scale Enhancement Pyramid Network for Small Object Detection from UAV Images.[J] Sun Jian;Gao Hongwei;Wang Xuna;Yu Jiahui Entropy (Basel; Switzerland) 2022,
  • [6] Focus-and-Detect: A small object detection framework for aerial images[J] Koyun Onur Can;Keser Reyhan Kevser;Akkaya İbrahim Batuhan;Töreyin Behçet Uğur Signal Processing: Image Communication 2022,
  • [7] CDNet: a real-time and robust crosswalk detection network on Jetson nano based on YOLOv5[J] Zhang Zheng De;Tan Meng Lu;Lan Zhi Cai;Liu Hai Chun;Pei Ling;Yu Wen Xian Neural Computing and Applications 2022,
  • [8] Enhancing Geometric Factors in Model Learning and Inference for Object Detection and Instance Segmentation.[J] Zheng Zhaohui;Wang Ping;Ren Dongwei;Liu Wei;Ye Rongguang;Hu Qinghua;Zuo Wangmeng IEEE transactions on cybernetics 2021,
  • [9] Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks.[J] Ren Shaoqing;He Kaiming;Girshick Ross;Sun Jian IEEE transactions on pattern analysis and machine intelligence 2017,
  • [10] Early and late mechanisms of surround suppression in striate cortex of macaque.[J] Webb Ben S;Dhruv Neel T;Solomon Samuel G;Tailby Chris;Lennie Peter The Journal of neuroscience : the official journal of the Society for Neuroscience 2005,