模型训练及关键点检测方法、装置和存储介质

本公开提出了一种模型训练及关键点检测方法、装置和存储介质，涉及装载自动化技术领域。其中，模型训练方法包括：利用关键点检测模型对包含待识别目标的样本图像进行处理，以得到特征图包含的多个网格单元中每个网格单元的关键点预测坐标的置信度向量、关键点预测类别的置信度向量；根据每个网格单元的关键点预测坐标的置信度向量，从多个网格单元中选取部分网格单元，作为待识别目标的每个关键点对应的正样本；根据=正样本的关键点预测坐标的置信度向量、关键点预测类别的置信度向量、每个关键点的真实坐标以及真实类别标签确定损失函数值，以对关键点检测模型进行训练。通过以上方法，能够提高关键点检测模型的训练稳定性和模型的检测精度。