基于稀疏Transformer的雷达点云三维目标检测

摘要：

随着计算机视觉技术的发展,基于点云的三维目标检测算法被广泛应用于自动驾驶、机器人控制等领域.针对点云稀疏条件下基于点云三维目标检测算法鲁棒性较差、检测精度低的问题,提出基于稀疏Transformer的三维目标检测算法.在注意力矩阵生成阶段,通过稀疏Transformer模块显式选择Top-t个权重元素,以保留有利于特征提取的权重元素,在降低环境噪点对鲁棒性影响的同时加快Transformer模块的运行速度.在回归阶段,将基于空间特征粗回归模块生成的边界框作为检测头模块的初始锚框,用于后续边界框的精细回归操作.设计基于体素的三维目标检测算法的损失函数,以精确地衡量类别损失、位置回归损失和方向损失.在KITTI数据集上的实验结果表明,相比PointPillars算法,该算法的平均精度均值提高3.46％,能有效提高点云三维目标的检测精度且具有较优的鲁棒性.相比原始Transformer模块,所提稀疏Transformer模块在点云图像上的平均运行速度加快了约0.54 frame/s.

关键词：机器视觉、三维目标检测、稀疏Transformer、粗回归、损失函数

所属期刊栏目：48

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

资助基金：上海市信息安全综合管理技术研究重点实验室开放项目AGK2019004

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

页数：共8页

页码：104-110,144