一种基于增强边缘特征的肺结节分割方法

本发明属于医学图像分割，具体涉及到一种增强边缘特征的肺结节分割方法。

背景技术：

1、肺癌是世界上最致命的癌症之一，患者在五年内生存率仅为19％。肺癌生存率与首次确诊时的疾病阶段密切相关，由于早期肺癌通常无症状，多数患者到中晚期才被诊断，只有21％的患者在早期被确诊。如能做到早期筛查，则患者五年内生存率将达到70％-90％，因此患者的早诊早治至关重要。目前，临床上以ct作为筛查肺癌的常规手段。

2、肺癌在ct上的早期表现通常为肺结节，是圆形或类圆形的致密影。肺结节分割是肺癌早期诊断的关键步骤，能够辅助放射科医生评估结节直径和形态。然而肺结节边界模糊、大小形态各异、纹理不均匀以及与肺实质中血管等组织相似，会对医生的判断产生干扰。因此，开发肺结节自动分割技术，可以提高肺癌临床诊断过程的效率，对后续诊疗具有重要意义。

3、近年来，许多深度学习方法被提出，并被应用于肺结节分割。深度学习模型利用结节切片进行训练，对于输入图像数据能够自动地学习并提取其特征，完成肺结节的分割任务。 u-net具有灵活、简单的优点，并且在少量样本下依旧能获得良好的分割效果，引起很多学者对其结构进行进一步研究。尽管很多模型在肺结节整体分割方面取得了较好效果，但在一定程度上忽略了结节模糊边缘纹理特征对分割结果的影响，致使模型分割精度不高。

技术实现思路

1、本发明的目的在于提供一种增强边缘特征的肺结节分割方法，使用该方法训练数据，解决现有的研究方法无法有效处理图像边缘纹理特征的问题。本发明提供的肺结节分割网络模型包括主干分支和边缘分支两个部分。其中，通过提供主干分支利用u-net提取结节全局信息，实现结节特征提取；通过提供gabor卷积模块作用于边缘分支，为充分提取结节边缘特征进行网络预测，解决u-net处理结节边缘纹理能力有限的问题。

2、本发明提供了一种增强边缘特征的肺结节分割方法，包括如下步骤：

3、步骤1：获取肺结节数据集，并对其进行数据预处理，构造出网络所需数据集，包括构造了肺结节原始数据集iimage、医生标注好的结节标签imask、边缘分支所需要的边缘标签iedge；

4、步骤2：对步骤1所获取的数据集iimage、imask和iedge分别进行划分以获得各自的训练集、验证集和测试集，对训练集和验证集进行数据扩充；

5、步骤3：提供一种用于肺结节分割任务的增强边缘特征双分支网络；

6、步骤4：使用步骤2中提供的训练集和验证集对提供的边缘特征增强网络进行训练；

7、步骤5：使用步骤4中训练好的网络对结节图像进行预测，并输出结节预测图像。

8、进一步地，所述步骤1中的肺结节数据集来自lidc-idri，根据标注文件中提供的结节位置中心点将原始图像与标签图像尺寸裁剪为96×96像素大小构造出iimage和imask，所述边缘标签iedge制作步骤具体如下：

9、步骤1.1：将结节标签记为f(x,y)，使用sobel算子获取图像梯度信息记为i(x,y)，可由式(1)来表达：

10、

11、其中ix代表sobel在x方向的算子与原图像f进行相关操作的结果，iy代表sobel在y方向的算子与原图像f进行相关操作的结果，i(x,y)代表图像在点(x,y)处梯度的幅值。

12、步骤1.2：对提取到的梯度值进行非极大值抑制；

13、步骤1.3：用双阈值算法检测和连接边缘，得到完整的边缘图像记作i′(x，y)；.

14、步骤1.4：遍历i′(x,y)得到边缘坐标信息，分别对检测到的每个边缘点扩展至三个像素点，即每个边缘点向左右两端延展一个单位像素，得到更加清晰的结节边缘标签图像iedge。

15、进一步地，所述步骤2中数据集的划分是将iimage、imask和iedge分别按照8：1：1 的比例划分训练集、验证集和测试集；数据扩充操作包括将iimage和imask中的图像进行随机裁剪、随机水平翻转、随机旋转。

16、进一步地，所述步骤3中提供的一种用于肺结节分割任务的增强边缘特征双分支网络包括主干网络和分支网络两个部分。

17、步骤3.1：提供具有u-net结构的主干分支。编码阶段由两个连续的3×3卷积来提取结节特征，并将relu激活函数替换为swish激活函数，将一个步长为2的2×2的最大池化层用于下采样，重复四次来降低图像尺寸获得结节浅层特征。解码阶段由一个2×2的反卷积和两个连续的3×3卷积组成，共经历四次来恢复图像尺寸并获取结节深层特征。在最后阶段，应用1×1卷积将图像的特征映射减少到所需通道数，生成分割图像。采用跳跃连接的方法，将编码器中提取到的结节特征拼接到解码器中对应的层加以结合获取更多结节信息。

18、步骤3.2：提供了具有编码-解码结构的边缘分支。编码部分使用gabor卷积模块，通过gabor卷积模块构成的编码器后，生成的特征图尺寸将减半，编码器共经历四次下采样获得所需特征图；解码器经历四次上采样将gabor卷积模块提取到的特征还原至输入尺寸。在边缘分支进行编码过程中采用跳跃连接的方式将提取到的纹理信息拼接到主干分支中进行整合，得到更为精细的分割结果。

19、进一步地，所述步骤3.2中的gabor卷积模块具体过程如下：

20、gabor卷积模块由两个卷积层、一个激活层、一个池化层以及一个通道注意力机制组成。gabor卷积模块通过一个3×3的卷积进行结节特征提取获取信息，随后使用一个gabor 卷积层增强网络对结节边缘特征的提取能力。利用最大池化层来降低图像分辨率，从而减少参数。进一步使用通道注意力机制来增强边缘纹理特征的响应，以增强边缘纹理特征的提取能力。

21、进一步地，所述步骤4中的对步骤2中提供的训练集、验证集对提供的边缘特征增强网络进行训练具体过程如下：

22、步骤4.1：提供网络训练参数，参数训练使用adam优化算法，批量大小设置为16，训练迭代100次，动量因子为0.9，初始学习率为0.05；

23、步骤4.2：提供混合损失函数l，由于网络模型是一种双分支结构，使用交叉熵损失函数来学习肺结节整体区域，focal loss作用于边缘分支以缓解样本极不平衡的情况，可以由式(2)来表达：

24、

25、其中，α代表权重因子，n代表样本像素点个数，yi代表像素点i的类别，正样本为1，负样本为0，γ代表动态缩放系数，pi代表像素点预测为正的概率。

26、进一步地，所述步骤5中使用步骤4中训练好的网络对结节图像进行预测是将图像输入至网络中，最后通过1×1卷积得到结节预测图像。

27、与现有技术相比，本发明有益效果如下：

28、1.本发明提供了一种对称双分支结构分割网络，主干分支通过利用u-net提取结节全局信息，由于u-net结构固定，不能提取不同尺度和方向上的特征，处理边缘纹理能力有限，故提供了gabor卷积模块作用于边缘分支来弥补u-net的缺点；

29、2.为充分提取结节边缘特征进行网络预测，本发明分析并制作了边缘标签；

30、3.在网络进行训练的过程中，网络中存在图像正负类样本不平衡的现象，为此本发明提供了一种混合损失函数，通过交叉熵作用于主干分支以及focal loss作用于边缘分支，将二者进行加权融合作为整个网络的损失来充分训练模型，进一步优化模型。