三维关键点预测及深度学习模型训练方法、装置及设备

摘要

本申请实施例提供了三维关键点预测及深度学习模型训练方法、装置及设备，该方法包括：获取包含待检测对象的RGB图像；基于RGB图像，获取待检测对象的二维关键点信息及每个二维关键点对应的热度图像；利用预先训练的深度学习模型，对RGB图像、二维关键点信息及热度图像进行关联分析，得到待检测对象的三维关键点信息的预测值。利用预先训练的深度学习模型对待检测对象的RGB图像、二维关键点信息及热度图像进行关联分析，相比于仅对二维关键点信息进行分析，能够从RGB图像及热度图像中获取除二维关键点信息外的其他用于预测三维关键点信息的数据，从而能够提高三维关键点信息预测的准确性。