基于Transformer的3D点云目标检测算法 |
| |
引用本文: | 刘明阳,杨啟明,胡冠华,郭岩,张建东.基于Transformer的3D点云目标检测算法[J].西北工业大学学报,2023(6):1190-1197. |
| |
作者姓名: | 刘明阳 杨啟明 胡冠华 郭岩 张建东 |
| |
作者单位: | 1. 沈阳飞机设计研究所;2. 西北工业大学电子信息学院;3. 中国船舶集团有限公司系统工程研究院 |
| |
基金项目: | 陕西省自然科学基础研究计划(2022JQ-593);;陕西省重点研发计划(2022GY-089)资助; |
| |
摘 要: | 针对在三维目标检测中由于空间维度的增加基于锚框的方法难以部署的问题,研究了基于集合预测的点云目标检测算法。提出一种基于Transformer的3D点云目标检测算法,并结合自动驾驶场景下的点云特点,提出了改进空间调制注意力和热图初始化策略进行训练加速和查询初始化,在浅层网络下取得了良好的检测性能。在KITTI数据集上与其他算法进行比较,结果表明所提算法在性能上已经达到先进水平,进一步对算法中的主要组成部分进行了消融实验,验证了各个模块对检测效果的贡献。
|
关 键 词: | Transformer 空间调制注意力机制 热图初始化 目标检测 深度学习 |
|
|