基于 YOLOv11n 的无人机航拍小目标检测算法学习

复制代码
基于 YOLOv11n 的无人机航拍小目标检测算法

问题:无人机航拍图像中小目标检测面临尺度变化大导致的检测精度较低和推理速度较慢等

解决:在 C3k2 模块中引入可变形卷积(DCN),增强模型在复杂背景下对

多尺度目标的特征提取能力,特别是对于形变显著的小目标。其次,针对无人机航拍图像中小目标占比较多等情形,新增 P2(Prediction

Layer 2)小目标检测层并去除了 P5(Prediction Layer 5)大目标层,有效提高了小目标特征的捕获精度,同时减少了大目标检测层的

冗余计算。然后,提出了一种结合 Focal Loss 与 DIoU Loss 的 Focal-DIoU(Focal Loss and Distance-IoU Loss)损失函数,解决了小目

标检测中的各目标类别不平衡问题,并优化了边界框的定位精度。最后,采用 LAMP 剪枝方法对模型进行轻量化处理,主要通过适当

调整剪枝加速比,自适应地按层评估并删除冗余连接,减少模型整体计算量并加速推理过程。

结果:文章所提 YOLOv11n-DPD 算法通过引

入可变形卷积、优化小目标检测层结构、改进损失函数和进行模型剪枝,提高了检测精度、加速了推理过程并优化了模型大小,与现

有的小目标检测方法相比表现更优,尤其适用于无人机航拍小目标检测任务和嵌入式平台的部署推理。

可变卷积过程

每个部分增加个偏移量

蛇形卷积

相关推荐
江畔柳前堤7 小时前
大语言模型分布式训练:从并行策略到万卡工程的系统梳理
人工智能·分布式·深度学习·算法·目标检测·机器学习·语言模型
江畔柳前堤8 小时前
HBM:大语言模型时代的「算力血液」——从内存墙到带宽革命的深度拆解
服务器·人工智能·windows·目标检测·语言模型·自然语言处理·软件工程
guo_xiao_xiao_13 小时前
YOLO人物肖像发型分类目标检测数据集
yolo·目标检测·分类
前网易架构师-高司机15 小时前
带标注的羽毛球运动员,裁判,羽毛球识别数据集,识别率84.4%,2879张图,支持yolo,coco json,voc xml,文末有模型训练代码
xml·yolo·json·数据集·羽毛球·裁判
guo_xiao_xiao_18 小时前
YOLO室内训练场橙色环形训练圈目标检测数据集-156张
人工智能·yolo·目标检测
FriendshipT1 天前
Ultralytics:解读 YOLO26 知识蒸馏
人工智能·pytorch·python·深度学习·yolo
动物园猫2 天前
无人机灾害场景人体目标检测数据集:10,000张图像 | 目标检测
人工智能·目标检测·无人机
向哆哆2 天前
铁路障碍物目标检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·目标检测
weixin_468466852 天前
目标检测精度上限与影响因素分析
图像处理·人工智能·目标检测·计算机视觉·图像分类·coco·检测精度
tangjunjun-owen2 天前
DETR系列发展演化全史(深度完整版)
目标检测·detr系列