摘要:针对自动驾驶场景下三维点云车辆的识别和定位问题, 提出了一种基于注意力机制的三维点云车辆目标检测算法. 算法将稀疏无序的点云空间划分成等距规则的体素表示, 用三维稀疏卷积和辅助网络同步从所有体素中提取内部点云特征, 进而生成鸟瞰图. 但在将内部三维的点云特征转化为二维的鸟瞰图后, 通常会造成目标空间特征信息丢失, 使得最终检测结果以及方向性预估差. 为进一步提取鸟瞰图中特征信息, 提出了一种注意力机制模块, 其中包含两种注意力模型, 并对其采用首、中、尾的“立体式”布局结构, 实现对鸟瞰图中特征信息的放大和抑制, 最后使用卷积神经网络和PS-Warp变换机制对处理过后的鸟瞰图进行三维目标检测. 实验表明, 该算法在保证实时检测效率的前提下, 与现有算法相比, 具有更好的方向预估性以及更高的检测精度.