基于注意力与特征融合的工程机械目标检测方法

摘要：

目的 针对施工环境中工程机械目标大小不一、相互遮挡、工作形态各异等问题,提出一种基于注意力与特征融合的目标检测方法(AT–FFRCNN).方法 在主干网络中采用ResNet50和特征路径聚合网络PFPN,融合不同尺度的特征信息,在区域建议网络(RPN)和全连接层引入注意力机制,提高目标识别的能力,在损失函数中使用广义交并比(GIoU),提高目标框的准确性.结果 实验表明,文中提出方法检测准确率比其他方法有较大提高,检测平均准确率(mAP)达到90％以上.结论 能够较好地完成工程机械目标的检测任务.

关键词：目标检测、Faster R—CNN、注意力机制、特征融合

所属期刊栏目：43

分类号：TP391(计算技术、计算机技术)

在线出版日期：2022-08-30（万方平台首次上网日期，不代表论文的发表时间）

页数：共7页

页码：61-67