过渡帧预测模型的训练方法以及过渡帧预测方法

摘要

本申请提供一种过渡帧预测模型的训练方法以及过渡帧预测方法，该训练方法包括：获取多张连续动作帧对应的目标关节数据，其中多张连续动作帧包括关键帧和中间帧；对中间帧的目标关节数据进行掩膜，获得中间帧的关节掩膜数据；将中间帧的关节掩膜数据和关键帧的目标关节数据作为自注意力神经网络模型的输入，其中自注意力神经网络模型是根据关节间从属关系构建的；调整自注意力神经网络模型的可学习参数，使自注意力神经网络模型输出的中间帧的关节预测结果与中间帧的目标关节数据的差异小于阈值，得到过渡帧预测模型。由此得到的过渡帧预测模型充分考虑了关节之间的联系，利用该模型预测得到的过渡帧，符合人体的运动规律，过渡帧的预测准确性高。