关键词:
英雄机器人
YOLOv5
DeepSort
嵌入式系统
摘要:
随着机器人技术发展,近年来移动机器人被应用到了生活中的各个领域,功能也日益强大,其中目标检测与跟踪技术是研究移动机器人的重点和难点,受到广泛研究者的关注。现实生活中移动机器人的工作领域复杂多变,移动机器人对目标进行检测跟踪时会受到光照变化、背景模糊、背景相似、目标被遮挡、目标发生形变、目标尺寸较小等因素的影响,加大工作难度。为解决上述问题,本文以机甲大师比赛为研究背景,在英雄机器人平台上提出一种融合改进YOLOv5的Deep Sort算法。该方法提高了英雄机器人对装甲板检测与跟踪的效果,进一步提升了英雄机器人对装甲板的打击精度。本文主要研究内容如下:
(1)英雄机器人采用嵌入式平台来实现对目标的检测,由于嵌入式平台的计算资源和内存空间有限,本文选取模型轻量化、检测精度高、易于部署的YOLOv5作为目标检测算法,但该算法在环境复杂和目标形变、目标尺寸较小等场景下的检测效果一般。为此,本文在YOLOv5主干网络的末端添加CA注意力机制,CA注意力将位置信息嵌入到通道注意力中,加强网络的特征提取能力,提升模型对装甲板的检测精度。采用SIo U损失函数替换YOLOv5中的GIo U损失函数,将目标框之间的向量角度特征引入目标检测的损失函数中,重新定义了惩罚项,有效地缓解了YOLOv5算法中训练和推理速度慢、准确率不高等问题。在模型中新增一个小目标检测层,减少模型对小目标的漏检率,提升模型的检测精度。在自行制作的数据集上对改进前后的算法进行实验对比,改进后的YOLOv5算法相比于原网络,精度提高了4.9%,召回率提高了0.8%,m AP@.5提升了5.3%,m AP@.5:.95提升了2.4%,结果表明改进后的YOLOv5算法性能更优。
(2)采集装甲板图片输入Deep Sort算法进行训练,将改进后的YOLOv5算法与Deep Sort算法进行融合,提高跟踪的精度。通过卡尔曼滤波进行轨迹预测,使用匈牙利算法将预测轨迹与当前帧的检测框级联匹配,解决目标之间发生重叠或遮挡时ID切换问题。最后进行卡尔曼滤波更新,实现对装甲板的跟踪。将融合改进YOLOv5的Deep Sort算法与原算法进行对比验证,改进后的Deep Sort算法相比于原网络跟踪准确度提升了2.1%,跟踪精度提升了2.6%。结果表明,改进后的算法在目标尺寸较小以及被遮挡情况下跟踪效果较好。
(3)将算法部署到Jetson TX2开发板上,使用Tensor RT对算法模型优化,加快系统的运行速度,并在英雄机器人平台上应用。经实验验证,本文改进算法能够满足英雄机器人实时检测与跟踪的需求,具有一定的可行性。