关键词:
YOLOv8
深度学习
点云定位
目标检测
Kinect传感器
摘要:
针对小目标检测任务中目标尺寸小,易受背景环境干扰,检测算法精度低等问题,本文提出了一种基于改进YOLOv8的小目标检测算法。该改进算法引入加权双向特征金字塔网络(Bidirectional Feature Pyramid Network,BiFPN)并融合P2小目标检测层,修改输出检测头位置,在训练模型时融合更多有效语义信息和位置信息,提高小目标检测的准确性。在Backbone中添加CBAM(Convolutional Block Attention Module)注意力机制模块,在Neck中添加自制G_SA注意力机制模块,结合通道注意力和空间注意力提高卷积神经网络的性能。采用Azure Kinect DK传感器,通过点云数据生成小目标的三维坐标信息,实现了目标的实时精准定位。实验结果表明,相较于原始YOLOv8n模型,该改进算法模型在VisDrone 2019数据集上的精准度(Precision)、召回率(Recall)和mAP50分别提高8%、9.2%和12%,验证了该算法的有效性。定位实验表明,该系统对于距离相机2.4米以内的小目标具有较高的定位精度,误差在0.4%以内。