基于 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 算法通过引

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

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

可变卷积过程

每个部分增加个偏移量

蛇形卷积

相关推荐
延凡科技3 小时前
多场景落地复盘:端边云架构无人机智能巡检系统设计与实践
大数据·数据结构·人工智能·科技·架构·无人机·能源
延卿7 小时前
nnDetection:基于 PyTorch 的目标检测框架
人工智能·pytorch·目标检测
科技大视界9 小时前
无人机智能巡检服务商技术哪家强?
无人机
前网易架构师-高司机10 小时前
带标注的细胞有丝分裂数据集数据集,识别率95.9%,304张图,支持yolo,coco json,voc xml,文末有模型训练代码
yolo·细胞·显微镜·生物·有丝分裂·显微
动物园猫13 小时前
PCB表面缺陷目标检测数据集:6类别、3,500张图像 | 目标检测
人工智能·目标检测·计算机视觉
前网易架构师-高司机1 天前
带标注的移动通信基站信号塔,可识别塔架和塔身数据集,识别率93.5%,1008张图,支持yolo,coco json,voc xml,文末有模型训练代码
yolo·数据集·通信基站·信号塔·塔架·塔身
AI棒棒牛1 天前
第11讲 | 目标检测任务:边界框、IoU、类别与置信度
人工智能·yolo·目标检测·yolo26
深度研习笔记1 天前
OpenCV工业视觉实战06|ROI区域截取+局部精细化检测,锁定作业区域、屏蔽无效背景,大幅提升YOLO检测准确率
人工智能·opencv·yolo
FriendshipT1 天前
Ultralytics:解读CBLinear模块
人工智能·pytorch·python·深度学习·目标检测
ShallWeL1 天前
Orin 上用 Docker 跑通目标检测功能
人工智能·目标检测·docker·容器