基于强化学习的无标签六维物体姿态预测方法及装置

摘要

本公开涉及人工智能技术领域，尤其涉及一种基于强化学习的无标签六维物体姿态预测方法及装置。所述技术方案包括：获取待预测的目标图像，所述目标图像为包括目标对象的二维图像；根据所述目标图像，采用预先训练得到的姿态预测模型进行姿态预测得到预测结果，所述姿态预测模型是根据样本图像进行强化学习得到的模型；根据所述预测结果，确定所述目标对象的三维位置和三维方向。本公开实施例通过引入强化学习来训练姿态预测模型，根据目标图像采用预先训练得到的姿态预测模型进行姿态预测，使得在没有真实的姿态标注的情况下可以解决基于二维图像的六维物体姿态估计的问题，保证了无标签六维物体姿态预测的预测效果。