有效

一种基于实例结构相关性的二维图像中三维目标检测方法

曹原周汉、李浥东、张慧、陈乃月、郎丛妍
北京交通大学
曹原周汉机构 暂无
技术领域 暂无
李浥东机构 暂无
技术领域 暂无
张慧机构 暂无
技术领域 暂无
陈乃月机构 暂无
技术领域 暂无
郎丛妍机构 暂无
技术领域 暂无

摘要

本发明提供了一种基于实例结构相关性的二维图像中三维目标检测方法。包括采集交通场景中的RGB图像集,利用RGB图像集构建训练集与测试集;构建基于卷积神经网络的三维目标检测模型,包括RGB图像特征提取主干网络、实例结构相关性构建模块和多任务检测分支,利用训练集对三维目标检测模型进行训练,通过随机梯度下降算法计算多任务损失函数,对三维目标检测模型中的参数进行更新,直至模型收敛,得到训练好的三维目标检测模型;利用训练好的三维目标检测模型对待检测二维RGB图像中的三维物体进行预测。本发明创新性地提取实例的深度结构模型,采用多头注意力机制构建实例之间的结构相关性,利用更新后的实例特征能够获得更准确的三维目标估计结果。