【技术实现步骤摘要】
乳腺腺管的图像分割方法、装置、电子设备以及存储介质
[0001]本申请涉及图像处理
,特别是涉及一种乳腺腺管的图像分割方法、装置、电子设备以及存储介质。
技术介绍
[0002]图像分割是指将图像中的感兴趣区域提取出来的技术和过程,它是由图像处理到图像分析的关键步骤。随着信息时代的发展,待处理的图片数量飞跃式的上升,利用深度学习网络进行图像分割的方法展现出了强大的性能以及便利性。深度学习网络通过不断的训练,提取图像的特征,分割出图像的感兴趣区域。因此,深度学习网络在图像分割领域有着广泛的研究价值和意义。
[0003]相关技术中,由于不同乳腺腺管之间形态差异显著,大小腺管面积差异高达百倍,深度学习网络对乳腺腺管区域的分割效果不明显,分割精度不高。
技术实现思路
[0004]基于此,本申请的目的在于,提供一种乳腺腺管的图像分割方法、装置、电子设备以及存储介质,其可提高乳腺腺管的图像分割精度。
[0005]根据本申请实施例的第一方面,提供一种乳腺腺管的图像分割方法,包括如下步骤:
[0006]获取乳腺腺管的样本图像数据集;
[0007]构建乳腺腺管的图像分割模型;乳腺腺管的图像分割模型包括依次连接的编码器、连接器以及解码器;编码器用于特征提取,连接器用于特征的权重计算,解码器用于特征融合;
[0008]根据样本图像数据集,训练编码器、连接器以及解码器,获得已训练的乳腺腺管的图像分割模型;
[0009]将待分割的乳腺腺管图像输入至已训练的乳腺腺管的图像分割模 ...
【技术保护点】
【技术特征摘要】
1.一种乳腺腺管的图像分割方法,其特征在于,所述方法包括如下步骤:获取乳腺腺管的样本图像数据集;构建乳腺腺管的图像分割模型;所述乳腺腺管的图像分割模型包括依次连接的编码器、连接器以及解码器;所述编码器用于特征提取,所述连接器用于特征的权重计算,所述解码器用于特征融合;根据所述样本图像数据集,训练所述编码器、所述连接器以及所述解码器,获得已训练的乳腺腺管的图像分割模型;将待分割的乳腺腺管图像输入至所述已训练的乳腺腺管的图像分割模型,获得乳腺腺管区域图像。2.根据权利要求1所述的乳腺腺管的图像分割方法,其特征在于:所述样本图像数据集包括若干个样本图像以及每个所述样本图像的分割标签;所述根据所述样本图像数据集,训练所述编码器、所述连接器以及所述解码器,获得已训练的乳腺腺管的图像分割模型的步骤,包括:将所述样本图像数据集中各样本图像输入至所述编码器,获得各所述样本图像的编码特征向量;将各所述样本图像的编码特征向量输入至所述连接器,获得各所述样本图像的连接特征向量;将各所述样本图像的连接特征向量和各所述样本图像的编码特征向量输入至所述解码器,获得各所述样本图像的分割结果;根据各所述样本图像的分割结果与分割标签,确定所述乳腺腺管的图像分割模型的损失函数;根据所述损失函数对所述乳腺腺管的图像分割模型进行迭代训练,直至满足预定训练终止条件,获得已训练的乳腺腺管的图像分割模型。3.根据权利要求2所述的乳腺腺管的图像分割方法,其特征在于:所述编码器包括若干个编码模块,每个所述编码模块均包括第一卷积层、第二卷积层、平均池化层、最大池化层以及第一感知机网络;所述将所述样本图像数据集中各样本图像输入至所述编码器,获得各所述样本图像的编码特征向量的步骤,包括:遍历每个所述编码模块,将所述样本图像数据集中各样本图像输入至当前所述编码模块的第一卷积层,获得各所述样本图像的第一特征向量;将各所述样本图像的第一特征向量输入至当前所述编码模块的平均池化层,获得各所述样本图像的第二特征向量;将各所述样本图像的第一特征向量输入至当前所述编码模块的最大池化层,获得各所述样本图像的第三特征向量;将各所述样本图像的第二特征向量以及第三特征向量输入至当前所述编码模块的第一感知机网络,获得各所述样本图像的第四特征向量;将各所述样本图像的第四特征向量与第一特征向量相乘,获得各所述样本图像的第五特征向量;对各所述样本图像的第五特征向量中各个向量元素计算平均值以及最大值,将所述平
均值以及所述最大值进行向量拼接,获得各所述样本图像的第六特征向量;将各所述样本图像的第六特征向量输入至当前所述编码模块的第二卷积层,获得各所述样本图像的第七特征向量;将各所述样本图像的第七特征向量与各所述样本图像的第五特征向量相乘,获得各所述样本图像的编码特征向量。4.根据权利要求2所述的乳腺腺管的图像分割方法,其特征在于:所述连接器包括第一线性拟合层、多头交叉注意力网络层、第二线性拟合层以及第二感知机网络;所述将各所述样本图像的编码特征向量输入至所述连接器,获得各所述样本图像的连接特征向量的步骤,包括:将各所述样本图像的编码特征向量输入至所述第一线性拟合层,获得各所述样本图像的第一线性特征向量;将各所述样本图像的第一线性特征向量输入至所述多头交叉注意力网络层,获得各所述样本图像的注意力特征向量;将各所述样本图像的注意力特征向量输入至所述第二线性拟合层,获得各所述样本图像的第二线性特征向量;将各所述样本图像的第二线性特征向量输入至所述第二感知机网络,获得各所述样本图像的第八特征向量;将各所述样本图像的第八特征向量与各所述样本图像的注意力特征向量相加,获得各所述样本图像的...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。