髋关节分割方法、装置、电子设备及计算机可读存储介质与流程

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

背景技术：

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

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

技术实现思路

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

2、第一方面，本技术实施例提供一种髋关节分割方法，包括：

3、获取待分割的髋关节图像；

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

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

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

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

8、获取髋关节图像数据集；

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

10、将每个髋关节图像及其对应的分割mask的图像格式均转换为png格式；

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

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

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

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

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

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

17、设置优化器为adam优化器；

18、设置损失函数为dice loss；

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

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

21、其中，残差卷积为由1x1，3x3和1x1卷积核组成的残差单元。

22、可选的，注意力机制网络包括位置注意力机制网络和通道注意力机制网络；

23、其中，位置注意力机制网络，用于通过所有位置处的特征的加权和来选择性地聚合每个位置的特征；

24、通道注意力机制网络，用于通过整合所有通道映射之间的相关特征来选择性地强调存在相互依赖的通道映射。

25、可选的，深度聚合金字塔网络以1/64图像分辨率的特征映射为输入，采用指数步长的大池核生成1/128、1/256、1/512图像分辨率的特征映射；

26、并且，利用全局平均池生成输入特征图和图像级信息，对特征图进行上采样，然后使用更多的3×3卷积，以分层残差的方式融合不同尺度的上下文信息，以获得更多的上下文信息。

27、第二方面，本技术实施例提供了一种髋关节分割装置，包括：

28、图像获取模块，用于获取待分割的髋关节图像；

29、髋关节分割结果获取模块，用于将髋关节图像输入预设的髋关节分割模型，输出髋关节分割结果；

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

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

32、可选的，装置还包括：

33、训练样本数据集获取模块，用于获取髋关节图像数据集；标注髋关节图像数据集中髋关节图像的髋关节区域，并将髋关节区域确定为分割mask；其中，每个分割mask与髋关节图像一一对应；将每个髋关节图像及其对应的分割mask的图像格式均转换为png格式；将所有转换为png格式的髋关节图像及其对应的分割mask，按照预设比例划分为训练集、验证集和测试集。

34、可选的，装置还包括：

35、模型训练模块，用于利用训练集对改进后的laddernet网络进行模型训练，并利用验证集和测试集进行验证和测试，得到髋关节分割模型。

36、可选的，模型训练模块，用于：

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

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

39、设置优化器为adam优化器；

40、设置损失函数为dice loss；

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

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

43、其中，残差卷积为由1x1，3x3和1x1卷积核组成的残差单元。

44、可选的，注意力机制网络包括位置注意力机制网络和通道注意力机制网络；

45、其中，位置注意力机制网络，用于通过所有位置处的特征的加权和来选择性地聚合每个位置的特征；

46、通道注意力机制网络，用于通过整合所有通道映射之间的相关特征来选择性地强调存在相互依赖的通道映射。

47、可选的，深度聚合金字塔网络以1/64图像分辨率的特征映射为输入，采用指数步长的大池核生成1/128、1/256、1/512图像分辨率的特征映射；

48、并且，利用全局平均池生成输入特征图和图像级信息，对特征图进行上采样，然后使用更多的3×3卷积，以分层残差的方式融合不同尺度的上下文信息，以获得更多的上下文信息。

49、第三方面，本技术实施例提供了一种电子设备，电子设备包括：处理器以及存储有计算机程序指令的存储器；

50、处理器执行计算机程序指令时实现如第一方面所示的髋关节分割方法。

51、第四方面，本技术实施例提供了一种计算机可读存储介质，计算机可读存储介质上存储有计算机程序指令，计算机程序指令被处理器执行时实现如第一方面所示的髋关节分割方法。

52、本技术实施例的髋关节分割方法、装置、电子设备及计算机可读存储介质，能够更加准确地进行髋关节分割。

53、该髋关节分割方法，包括：获取待分割的髋关节图像；将髋关节图像输入预设的髋关节分割模型，输出髋关节分割结果；

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

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

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

57、并且，在laddernet网络的最后一层添加深度聚合金字塔网络，用于提取不同层级不同尺度的卷积特征，挖掘髋关节图像最后一层的特征信息，以定位分割位置，故能够更加准确地进行髋关节分割。