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

被引:7
|
作者
吴旭红 [1 ]
赵清华 [1 ]
机构
[1] 太原理工大学
关键词
YOLOv7; 无人机; 航拍图像; 小目标检测; SimAM注意力机制;
D O I
暂无
中图分类号
V279 [无人驾驶飞机]; TP183 [人工神经网络与计算]; TP391.41 [];
学科分类号
080203 ; 081104 ; 0812 ; 0835 ; 1111 ; 1405 ;
摘要
针对无人机捕获场景下目标尺度变化剧烈、小目标检测精度低、漏检率高等问题,提出了一种改进的YOLOv7目标检测算法。首先,在原YOLOv7基础上增加一个极小目标检测层,以适应不同尺度下的物体目标,降低小目标的漏检率;其次,在特征融合网络中引入无参注意力机制,并基于该注意力机制构建了一个MP-SimAM模块,使网络融合更多重要的特征信息;最后,提出了一种新的边框回归损失函数SCIoU Loss,进一步提升模型的收敛速度与检测精度。实验结果表明,该模型在VisDrone 2019数据集上表现出色,所提算法模型在测试集上mAP50达44.0%,相比于基准模型YOLOv7提升了2.6个百分点,对于小目标的检测效果提升明显。
引用
收藏
页码:35 / 40 +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, 11
  • [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
    Zhang, Zheng-De
    Tan, Meng-Lu
    Lan, Zhi-Cai
    Liu, Hai-Chun
    Pei, Ling
    Yu, Wen-Xian
    [J]. NEURAL COMPUTING & APPLICATIONS, 2022, 34 (13): : 10719 - 10730
  • [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
    Ren, Shaoqing
    He, Kaiming
    Girshick, Ross
    Sun, Jian
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (06) : 1137 - 1149
  • [10] Early and late mechanisms of surround suppression in striate cortex of macaque
    Webb, BS
    Dhruv, NT
    Solomon, SG
    Tailby, C
    Lennie, P
    [J]. JOURNAL OF NEUROSCIENCE, 2005, 25 (50): : 11666 - 11675