[发明专利]膝关节分割方法、装置、电子设备及计算机可读存储介质

说明书

本申请提供了一种膝关节分割方法、装置、电子设备及计算机可读存储介质。该膝关节分割方法，包括：获取待分割的膝关节图像；将膝关节图像输入预设的膝关节分割模型，输出膝关节分割结果；其中，膝关节分割模型是基于改进后的FADNet网络经过模型训练得到的，改进后的FADNet网络是在FADNet网络的每层每对相邻的解码器分支和编码器分支之间的跳跃连接中添加注意力机制网络，用于增强对膝关节图像边界特征的分割；并且，在FADNet网络的最后一层添加深度聚合金字塔网络，用于提取不同层级不同尺度的卷积特征，挖掘膝关节图像最后一层的特征信息，以定位分割位置。根据本申请实施例，能够更加准确地进行膝关节分割。

技术领域

本申请属于深度学习智能识别技术领域，尤其涉及一种膝关节分割方法、装置、电子设备及计算机可读存储介质。

背景技术

传统图像分割算法是基于阈值分割、边缘检测或区域增长来分割医学图像，这类方法分割精度较差，粗略分割目标区域，不能精确分割目标区域边界。

因此，如何更加准确地进行膝关节分割是本领域技术人员亟需解决的技术问题。

发明内容

本申请实施例提供一种膝关节分割方法、装置、电子设备及计算机可读存储介质，能够更加准确地进行膝关节分割。

第一方面，本申请实施例提供一种膝关节分割方法，包括：

获取待分割的膝关节图像；

将膝关节图像输入预设的膝关节分割模型，输出膝关节分割结果；

其中，膝关节分割模型是基于改进后的FADNet网络经过模型训练得到的，改进后的FADNet网络是在FADNet网络的每层每对相邻的解码器分支和编码器分支之间的跳跃连接中添加注意力机制网络，用于增强对膝关节图像边界特征的分割；

并且，在FADNet网络的最后一层添加深度聚合金字塔网络，用于提取不同层级不同尺度的卷积特征，挖掘膝关节图像最后一层的特征信息，以定位分割位置。

可选的，在将膝关节图像输入预设的膝关节分割模型之前，方法还包括：

获取膝关节图像数据集；

标注膝关节图像数据集中膝关节图像的膝关节区域，并将膝关节区域确定为分割mask；其中，每个分割mask与膝关节图像一一对应；

将每个膝关节图像及其对应的分割mask的图像格式均转换为PNG格式；

将所有转换为PNG格式的膝关节图像及其对应的分割mask，按照预设比例划分为训练集、验证集和测试集。

可选的，在将所有转换为PNG格式的膝关节图像及其对应的分割mask，按照预设比例划分为训练集、验证集和测试集之后，方法还包括：

利用训练集对改进后的FADNet网络进行模型训练，并利用验证集和测试集进行验证和测试，得到膝关节分割模型。

可选的，利用训练集对改进后的FADNet网络进行模型训练，并利用验证集和测试集进行验证和测试，得到膝关节分割模型，包括：

在模型训练过程中，设置训练的batch_size为64；

设置初始化学习率为1e-4，附加学习率衰减策略，每迭代5000次，学习率衰减为上一次学习率的0.9；

设置优化器为Adam优化器；

设置损失函数为DICE loss；

设置每迭代1000次，对训练集和验证集做一次验证，通过早停法来判断网络训练停止时间，并得到膝关节分割模型。

可选的，FADNet网络的卷积层使用残差卷积，以减少特征损失；