基于深度学习的关节影像关键点自动检测方法、装置、设备及存储介质

摘要

本发明提供了一种基于深度学习的关节影像关键点自动检测方法、装置、设备及存储介质。该方法包括：获取样本图像，所述样本图像包括关键点位置的标注信息；将所述样本图像输入预设的关键点检测模型，所述关键点检测模型包括U‑Net网络；利用所述预设的关键点检测模型对所述样本图像进行关键点检测，得到所述样本图像的预测关键点的第一热力图；根据所述样本图像的预测关键点的第一热力图和设定的损失函数对所述预设的关键点检测模型的网络权值进行训练，得到目标关键点检测模型。本发明的技术能够实现关节影像中关键点的自动提取，提高关键点检测的一致性，降低人为因素的影响。