基于深度学习卷积神经网络的自动分割前列腺全腺体方法转让专利

申请号 : CN202210096760.1

文献号 : CN114399501B

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 许梨梨孙昊张古沐阳李秀丽毛丽张晓霄白鑫陈丽张家慧金征宇

申请人 : 中国医学科学院北京协和医院

摘要 :

本发明涉及基于深度学习卷积神经网络的自动分割前列腺全腺体方法。包括:获取样本的前列腺MR图像;将所述前列腺MR图像输入前列腺全腺体分割网络,得到前列腺全腺体图像,所述前列腺全腺体分割网络包括前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述前列腺尖部分割网络用于分割前列腺MR图像得到前列腺尖部图像,所述前列腺中部分割网络用于分割前列腺MR图像得到前列腺中部图像,所述前列腺底部分割网络用于分割前列腺MR图像得到前列腺底部图像,将所述前列腺尖部图像、前列腺中部图像和前列腺底部图像进行图像合并,得到前列腺全腺体分割图像。本发明从前列腺解剖结构特点出发,提供更好的前列腺全腺体分割方法。

权利要求 :

1.一种基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,具体方法步骤包括:

获取样本的前列腺MR图像;

将前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像;

将所述T2WI图像输入第一前列腺全腺体分割网络,得到第一前列腺全腺体图像,所述第一前列腺全腺体分割网络包括第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络;将所述DWI图像输入第二前列腺全腺体分割网络,得到第二前列腺全腺体图像,所述第二前列腺全腺体分割网络包括第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络;将所述ADC图像输入第三前列腺全腺体分割网络,得到第三前列腺全腺体图像,所述第三前列腺全腺体分割网络包括第三前列腺尖部分割网络、第三前列腺中部分割网络和第三前列腺底部分割网络;将所述DCE图像输入第四前列腺全腺体分割网络,得到第四前列腺全腺体图像,所述第四前列腺全腺体分割网络包括第四前列腺尖部分割网络、第四前列腺中部分割网络和第四前列腺底部分割网络。

2.根据权利要求1中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,所述将前列腺MR图像进行分类为将所述前列腺MR图像输入分类器中,所述分类器基于所述前列腺MR图像的分辨率或序列名称对前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像。

3.根据权利要求1中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,所述方法步骤还包括将前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像。

4.根据权利要求1中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,将所述前列腺MR图像输入分类器中,所述分类器基于前列腺解剖结构对前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像。

5.根据权利要求1‑4任意一项中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,采用分类器对前列腺MR图像进行分类,所述分类器选自下列机器学习模型算法中的一种或几种:逻辑回归算法、朴素贝叶斯分类、支持向量机、k近邻、决策树、随机森林、xgboost、感知机算法。

6.根据权利要求1中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,所述分割网络为3D U‑net分割网络。

7.根据权利要求6中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,所述3D U‑net分割网络包括3‑5个上采样模块和3‑5个下采样模块,所述第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络为包括5个上采样模块和5个下采样模块;所述第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络为包括3个上采样模块和3个下采样模块。

8.根据权利要求1中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,所述前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络为预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述预先训练包括:获取训练样本的前列腺MR图像;

对所述前列腺MR图像进行数据增强得到数据增强处理后的前列腺MR图像;将所述数据增强处理后前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;

将所述前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像分别输入前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,得到前列腺尖部图像、前列腺中部图像和前列腺底部图像,将得到的前列腺尖部图像、前列腺中部图像和前列腺底部图像与医生手动圈定的图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络。

9.根据权利要求8中所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法,其特征在于,所述数据增强包括水平镜像、随机裁剪、平移、亮度增强、对比度增强中的一种或几种。

10.一种基于深度学习卷积神经网络的自动分割前列腺全腺体的系统,其特征在于,包括:

获取单元,用于获取样本的前列腺MR图像;

分割单元,用于将前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像;将所述T2WI图像输入第一前列腺全腺体分割网络,得到第一前列腺全腺体图像,所述第一前列腺全腺体分割网络包括第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络;将所述DWI图像输入第二前列腺全腺体分割网络,得到第二前列腺全腺体图像,所述第二前列腺全腺体分割网络包括第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络;将所述ADC图像输入第三前列腺全腺体分割网络,得到第三前列腺全腺体图像,所述第三前列腺全腺体分割网络包括第三前列腺尖部分割网络、第三前列腺中部分割网络和第三前列腺底部分割网络;将所述DCE图像输入第四前列腺全腺体分割网络,得到第四前列腺全腺体图像,所述第四前列腺全腺体分割网络包括第四前列腺尖部分割网络、第四前列腺中部分割网络和第四前列腺底部分割网络。

11.一种基于深度学习卷积神经网络的自动分割前列腺全腺体的设备,其特征在于,包括:存储器和处理器;

所述存储器用于存储程序指令;

所述处理器用于调用程序指令,当程序指令被执行时实现权利要求1‑9任意一项所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法步骤。

12.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1‑9任意一项所述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法步骤。

说明书 :

基于深度学习卷积神经网络的自动分割前列腺全腺体方法

技术领域

[0001] 本发明涉及智能医疗技术领域,更具体地,涉及一种基于深度学习卷积神经网络的自动分割前列腺全腺体方法、设备、系统及存储介质。

背景技术

[0002] MR 图像上准确的前列腺全腺分割在前列腺癌甚至良性前列腺增生的管理中起着重要作用,对于恶性前列腺癌,准确高效的前列腺轮廓识别对癌症分期和放射治疗计划等至关重要。深度学习的进步在医学影像分析方面取得了突出的成果,最近提出的 U‑Net 架构已成功应用于前列腺分割。然而,这些研究大多基于公共数据集或单一图像,有些分割模型的效果泛化性不佳,这些很大原因是由于没有深入考虑前列腺MR图像中各种图像的分辨率差异较大,没有从前列腺解剖结构出发解决这些临床影像实际情况。

发明内容

[0003] 为解决上述问题,我们基于前列腺MR图像的特点和前列腺解剖结构特点,建立一种基于深度学习卷积神经网络的自动分割前列腺全腺体方法、设备、系统及存储介质。
[0004] 本申请公开了一种基于深度学习卷积神经网络的自动分割前列腺全腺体方法,具体方法步骤包括:
[0005] 获取样本的前列腺MR图像;
[0006] 将所述前列腺MR图像输入前列腺全腺体分割网络,得到前列腺全腺体图像,所述前列腺全腺体分割网络包括前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述前列腺尖部分割网络用于分割前列腺MR图像得到前列腺尖部图像,所述前列腺中部分割网络用于分割前列腺MR图像得到前列腺中部图像,所述前列腺底部分割网络用于分割前列腺MR图像得到前列腺底部图像,将所述前列腺尖部图像、前列腺中部图像和前列腺底部图像进行图像合并,得到前列腺全腺体分割图像。
[0007] 进一步,所述方法步骤还包括将前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像;可选的,将所述前列腺MR图像输入分类器中,所述分类器基于所述前列腺MR图像信息中的序列名称或分辨率对前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像。
[0008] 进一步,所述方法步骤还包括将前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;可选的,将所述前列腺MR图像输入分类器中,所述分类器基于前列腺解剖结构对前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像。
[0009] 进一步,采用分类器对前列腺MR图像进行分类,所述分类器选自下列机器学习模型算法中的一种或几种:逻辑回归算法、朴素贝叶斯分类、支持向量机、k近邻、决策树、随机森林、xgboost、感知机算法。
[0010] 进一步,将所述T2WI图像输入第一前列腺全腺体分割网络,得到第一前列腺全腺体图像,所述第一前列腺全腺体分割网络包括第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络;将所述DWI图像输入第二前列腺全腺体分割网络,得到第二前列腺全腺体图像,所述第二前列腺全腺体分割网络包括第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络;将所述ADC图像输入第三前列腺全腺体分割网络,得到第三前列腺全腺体图像,所述第三前列腺全腺体分割网络包括第三前列腺尖部分割网络、第三前列腺中部分割网络和第三前列腺底部分割网络; 将所述DCE图像输入第四前列腺全腺体分割网络,得到第四前列腺全腺体图像,所述第四前列腺全腺体分割网络包括第四前列腺尖部分割网络、第四前列腺中部分割网络和第四前列腺底部分割网络。
[0011] 进一步,所述分割网络为3D U‑net分割网络;可选的,所述3D U‑net分割网络包括3‑5个上采样模块和3‑5个下采样模块,所述第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络为包括5个上采样模块和5个下采样模块;所述第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络包括3个上采样模块和3个下采样模块。进一步,所述前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络为预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述预先训练包括:
[0012] 获取训练样本的前列腺MR图像;
[0013] 对所述前列腺MR图像进行数据增强得到数据增强处理后的前列腺MR图像,可选的,所述数据增强包括水平镜像、随机裁剪、平移、亮度增强、对比度增强中的一种或几种;
[0014] 将所述数据增强处理后前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;
[0015] 将所述前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像分别输入前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,得到前列腺尖部图像、前列腺中部图像和前列腺底部图像,将得到的前列腺尖部图像、前列腺中部图像和前列腺底部图像与医生手动圈定的图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络。
[0016] 本申请的目的在于提供一种基于深度学习卷积神经网络的自动分割前列腺全腺体的系统,包括:
[0017] 获取单元,用于获取样本的前列腺MR图像;
[0018] 分割单元,用于将所述前列腺MR图像输入前列腺全腺体分割网络,得到前列腺全腺体图像,所述前列腺全腺体分割网络包括前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述前列腺尖部分割网络用于分割前列腺MR图像得到前列腺尖部图像,所述前列腺中部分割网络用于分割前列腺MR图像得到前列腺中部图像,所述前列腺底部分割网络用于分割前列腺MR图像得到前列腺底部图像,将所述前列腺尖部图像、前列腺中部图像和前列腺底部图像进行合并,得到前列腺全腺体分割图像。
[0019] 本申请的目的在于提供一种基于深度学习卷积神经网络的自动分割前列腺全腺体的设备,包括:存储器和处理器;
[0020] 所述存储器用于存储程序指令;
[0021] 所述处理器用于调用程序指令,当程序指令被执行时实现上述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法步骤。
[0022] 一种计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现上述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法步骤。
[0023] 本申请的优点:
[0024] 1.本申请考虑到前列腺MR图像中T2WI图像、DWI图像、ADC图像或DCE图像的分辨率差异较大,将不同种图像样本同时用于模型训练时将导致模型训练结果不理想,故而在进行分割网络训练时,分别进行了分割网络模型的训练,并且在分割网络上采样模块和下采样模块上分别进行了调整,调整后的模型能自动准确的分割T2WI图像、DWI图像、ADC图像或DCE图像前列腺整个腺体;
[0025] 2.本申请在实际模型训练中发现部分区域分割效果不佳,考虑前列腺解剖结构差异,将前列腺全腺体MR图像分为前列腺尖部MR图像、前列腺中部MR图像和前列腺底部MR图像,分别训练前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,尤其是对前列腺尖部分割网络,我们在模型训练中发现其相较中部分割网络和底部分割网络,容易受图像扫描参数的影响,为此,在模型训练中增加图像的预处理,包括前列腺尖部MR图像的加噪和降噪等处理,提升模型的泛化能力。

附图说明

[0026] 为了更清楚地说明本发明实施例中的技术方案,下面将对实施例描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获取其他的附图。
[0027] 图1是本发明实施例提供的一种基于深度学习卷积神经网络的自动分割前列腺全腺体方法示意流程图;
[0028] 图2是本发明实施例提供的一种基于深度学习卷积神经网络的自动分割前列腺全腺体系统示意图;
[0029] 图3是本发明实施例提供的一种基于深度学习卷积神经网络的自动分割前列腺全腺体设备示意图。

具体实施方式

[0030] 为了使本技术领域的人员更好地理解本发明方案,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述。
[0031] 在本发明的说明书和权利要求书及上述附图中的描述的一些流程中,包含了按照特定顺序出现的多个操作,但是应该清楚了解,这些操作可以不按照其在本文中出现的顺序来执行或并行执行,操作的序号如101、102等,仅仅是用于区分开各个不同的操作,序号本身不代表任何的执行顺序。另外,这些流程可以包括更多或更少的操作,并且这些操作可以按顺序执行或并行执行。需要说明的是,本文中的“第一”、“第二”等描述,是用于区分不同的消息、设备、模块等,不代表先后顺序,也不限定“第一”和“第二”是不同的类型。
[0032] 下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域技术人员在没有作出创造性劳动前提下所获取的所有其他实施例,都属于本发明保护的范围。
[0033] 图1是本发明实施例提供的一种基于深度学习卷积神经网络的自动分割前列腺全腺体方法示意流程图,具体地,所述方法包括如下步骤:
[0034] 101:获取样本的前列腺MR图像;
[0035] 在一个实施例中,所述前列腺MR图像包括扩散加权成像 (DWI) 、 T2 加权成像 (T2WI) 图像、ADC图像和DCE图像。可选的,所述T2WI 图像采用不同类型扫描器获得,例如T2WI 图像采用GE Discovery MR 750、Siemens Magnetom Skyra、Siemens Magnetom TrioTim、Siemens Magnetom Skyra、Siemens Magnetom Aera、Siemens Magnetom Espree、GE Optima MR360、GE Signa EXCITE、Philips Achieva、GE SIGNA Pioneer等扫描器获得,例如DWI 图像采用GE Discovery MR 750、Siemens Magnetom Skyra、Siemens Magnetom TrioTim等扫描器获得。
[0036] 在一个具体实施例中,所述方法步骤还包括获取样本的前列腺MR图像后,将前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像;可选的,将所述前列腺MR图像输入分类器中,所述分类器基于所述前列腺MR图像的分辨率或序列名称对前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像。可选的,采用分类器对前列腺MR图像进行分类,所述分类器选自下列机器学习模型算法中的一种或几种:逻辑回归算法、朴素贝叶斯分类、支持向量机、k近邻、决策树、随机森林、xgboost、感知机算法。
[0037] 在一个具体实施例中,分类器基于所述前列腺MR图像的序列名称对前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像。更具体的,分类器自动读取Dicom的影像信息,提取扫描序列名称“Series Description”这一关键字,例如 T2WI图像的序列名称通常包含“T2”,DWI图像的序列名称通常包含“DWI”,将所述前列腺MR图像分为T2WI图像或DWI图像。
[0038] 在一个具体实施例中,所述方法步骤还包括将前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;可选的,将所述前列腺MR图像输入分类器中,所述分类器基于前列腺解剖结构对前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像。可选的,采用分类器对前列腺MR图像进行分类,所述分类器选自下列机器学习模型算法中的一种或几种:逻辑回归算法、朴素贝叶斯分类、支持向量机、k近邻、决策树、随机森林、xgboost、感知机算法。
[0039] 在一个实施例中,所述方法步骤还包括获取样本的前列腺MR图像后,对MR图像进行预处理,所述预处理包括归一化处理等。在一个具体实施例中,将图像重新采样到统一的像素间距,即训练队列像素间距的中值,以抵消分辨率不一致导致的偏差(T2WI 为 3.69*0.51*0.51,3.21*1.41*1.41 DWI)。图像的输入块大小是所有图像的平均非零区域(T2WI 为 14*352*352, DWI 为 20*174*250)。然后通过 z‑score 归一化对图像的强度进行归一化。
[0040] 102:将所述前列腺MR图像输入前列腺全腺体分割网络,得到前列腺全腺体图像,所述前列腺全腺体分割网络包括前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述前列腺尖部分割网络用于分割前列腺MR图像得到前列腺尖部图像,所述前列腺中部分割网络用于分割前列腺MR图像得到前列腺中部图像,所述前列腺底部分割网络用于分割前列腺MR图像得到前列腺底部图像,将所述前列腺尖部图像、前列腺中部图像和前列腺底部图像进行图像合并,得到前列腺全腺体分割图像。
[0041] 在一个实施例中,将所述T2WI图像输入第一前列腺全腺体分割网络,得到第一前列腺全腺体图像,所述第一前列腺全腺体分割网络包括第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络,所述第一前列腺尖部分割网络用于分割前列腺MR图像得到第一前列腺尖部图像,所述第一前列腺中部分割网络用于分割前列腺MR图像得到第一前列腺中部图像,所述第一前列腺底部分割网络用于分割前列腺MR图像得到第一前列腺底部图像,将所述第一前列腺尖部图像、第一前列腺中部图像和第一前列腺底部图像进行合并,得到T2WI图像前列腺全腺体分割图像;将所述DWI图像输入第二前列腺全腺体分割网络,得到第二前列腺全腺体图像,所述第二前列腺全腺体分割网络包括第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络,所述第二前列腺尖部分割网络用于分割前列腺MR图像得到第二前列腺尖部图像,所述第二前列腺中部分割网络用于分割前列腺MR图像得到第二前列腺中部图像,所述第二前列腺底部分割网络用于分割前列腺MR图像得到第二前列腺底部图像,将所述第二前列腺尖部图像、第二前列腺中部图像和第二前列腺底部图像进行合并,得到DWI图像前列腺全腺体分割图像。
[0042] 在一个实施例中,所述分割网络为3D U‑net分割网络;可选的,所述3D U‑net分割网络包括3‑5个上采样模块和3‑5个下采样模块,所述第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络为包括5个上采样模块和5个下采样模块;所述第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络为包括3个上采样模块和3个下采样模块。在一个具体实施例中,3D U‑net分割网络每个下采样模块包含卷积层、Batch Normalization、ReLU、max_polling,每个上采样模块包含上采样层、卷积层、Batch Normalization、ReLU。在一个具体实施例中,3D U‑net分割网络训练时损失函数是 Dice 损失和二元交叉熵 (BCE) 损失的组合。
[0043] 在一个实施例中,所述图像合并为将三个分割网络(前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络)的预测值相加,其中背景预测值除以3,然后根据相加后的预测值来生成全腺体分割结果。
[0044] 在一个实施例中,所述前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络为预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述预先训练包括:获取训练样本的前列腺MR图像;对所述前列腺MR图像进行数据增强得到数据增强处理后的前列腺MR图像,可选的,所述数据增强包括水平镜像、随机裁剪、平移、亮度增强、对比度增强中的一种或几种;将所述数据增强处理后前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;将所述前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像分别输入前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,得到前列腺尖部图像、前列腺中部图像和前列腺底部图像,将得到的前列腺尖部图像、前列腺中部图像和前列腺底部图像与医生手动圈定的图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络。
[0045] 在一个实施例中,,所述预先训练包括:获取训练样本的前列腺MR图像;将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像,对所述前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像进行数据增强得到数据增强处理后的前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;将所述数据增强处理后的前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像分别输入前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,得到前列腺尖部图像、前列腺中部图像和前列腺底部图像,将得到的前列腺尖部图像、前列腺中部图像和前列腺底部图像与医生手动圈定的图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络。可选的,将所述前列腺尖部MR图像和数据增强处理后的前列腺尖部MR图像输入前列腺尖部分割网络。
[0046] 在一个实施例中,所述预先训练好的前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络是通过如下步骤训练得到的:获取训练样本的前列腺MR图像;将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;对所述前列腺尖部MR图像进行数据增强得到数据增强处理后的前列腺MR图像,所述数据增强包括水平镜像、随机裁剪、平移、亮度增强、对比度增强中的一种或几种,优选的,所述数据增强还包括图像加噪和图像降噪;将所述前列腺MR图像和增强处理后前列腺尖部MR图像输入前列腺尖部分割网络,得到前列腺尖部图像,将得到的前列腺尖部图像与医生手动圈定的前列腺尖部图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺尖部分割网络;通过引入增强处理后的前列腺尖部MR图像,如加噪的前列腺尖部MR图像、降噪的前列腺尖部MR图像和原前列腺尖部MR图像一起作为训练集进行前列腺尖部分割网络,使得训练得到的前列腺尖部分割网络具有更好的泛化性。
[0047] 对所述前列腺中部MR图像进行数据增强得到数据增强处理后的前列腺MR图像,所述数据增强包括水平镜像、随机裁剪、平移、亮度增强、对比度增强中的一种或几种;将所述增强处理后前列腺中部MR图像输入前列腺中部分割网络,得到前列腺中部图像,将得到的前列腺中部图像与医生手动圈定的前列腺中部图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺中部分割网络;
[0048] 对所述前列腺底部MR图像进行数据增强得到数据增强处理后的前列腺MR图像,所述数据增强包括水平镜像、随机裁剪、平移、亮度增强、对比度增强中的一种或几种;将所述增强处理后前列腺底部MR图像输入前列腺中部分割网络,得到前列腺底部图像,将得到的前列腺底部图像与医生手动圈定的前列腺底部图像进行比对,生成损失值,反向传播,优化网络参数,得到预先训练好的前列腺底部分割网络。
[0049] 在一个实施例中,使用Dice similarity coefficient (DSC)、Hausdorff distance (95HD, mm)、Average boundary distance (ABD, mm)来评估3D U‑net分割网络的性能。DSC 被广泛用于量化分割之间的空间重叠,其值范围从 0(表示没有重叠)到 1(表示完全重叠)。
[0050] 图2是本发明实施例提供的一种基于深度学习卷积神经网络的自动分割前列腺全腺体的系统,包括:
[0051] 获取单元201,用于获取样本的前列腺MR图像;
[0052] 分割单元202,用于将所述前列腺MR图像输入前列腺全腺体分割网络,得到前列腺全腺体图像,所述前列腺全腺体分割网络包括前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,所述前列腺尖部分割网络用于分割前列腺MR图像得到前列腺尖部图像,所述前列腺中部分割网络用于分割前列腺MR图像得到前列腺中部图像,所述前列腺底部分割网络用于分割前列腺MR图像得到前列腺底部图像,将所述前列腺尖部图像、前列腺中部图像和前列腺底部图像进行合并,得到前列腺全腺体分割图像。
[0053] 在一个具体实施例中,所述基于深度学习卷积神经网络的自动分割前列腺全腺体的系统,包括。
[0054] 获取单元,用于获取样本的前列腺MR图像;
[0055] 分类单元,用于将前列腺MR图像进行分类,将所述前列腺MR图像分为T2WI图像、DWI图像、ADC图像或DCE图像;
[0056] 分割单元,用于将所述T2WI图像输入第一前列腺全腺体分割网络,得到第一前列腺全腺体图像,所述第一前列腺全腺体分割网络包括第一前列腺尖部分割网络、第一前列腺中部分割网络和第一前列腺底部分割网络,所述第一前列腺尖部分割网络用于分割前列腺MR图像得到第一前列腺尖部图像,所述第一前列腺中部分割网络用于分割前列腺MR图像得到第一前列腺中部图像,所述第一前列腺底部分割网络用于分割前列腺MR图像得到第一前列腺底部图像,将所述第一前列腺尖部图像、第一前列腺中部图像和第一前列腺底部图像进行合并,得到T2WI图像前列腺全腺体分割图像;
[0057] 将所述DWI图像输入第二前列腺全腺体分割网络,得到第二前列腺全腺体图像,所述第二前列腺全腺体分割网络包括第二前列腺尖部分割网络、第二前列腺中部分割网络和第二前列腺底部分割网络,所述第二前列腺尖部分割网络用于分割前列腺MR图像得到第二前列腺尖部图像,所述第二前列腺中部分割网络用于分割前列腺MR图像得到第二前列腺中部图像,所述第二前列腺底部分割网络用于分割前列腺MR图像得到第二前列腺底部图像,将所述第二前列腺尖部图像、第二前列腺中部图像和第二前列腺底部图像进行合并,得到DWI图像前列腺全腺体分割图像;
[0058] 将所述ADC图像输入第三前列腺全腺体分割网络,得到第三前列腺全腺体图像,所述第三前列腺全腺体分割网络包括第三前列腺尖部分割网络、第三前列腺中部分割网络和第三前列腺底部分割网络,所述第三前列腺尖部分割网络用于分割前列腺MR图像得到第三前列腺尖部图像,所述第三前列腺中部分割网络用于分割前列腺MR图像得到第三前列腺中部图像,所述第三前列腺底部分割网络用于分割前列腺MR图像得到第三前列腺底部图像,将所述第三前列腺尖部图像、第三前列腺中部图像和第三前列腺底部图像进行合并,得到ADC图像前列腺全腺体分割图像;
[0059] 将所述DCE图像输入第四前列腺全腺体分割网络,得到第四前列腺全腺体图像,所述第四前列腺全腺体分割网络包括第四前列腺尖部分割网络、第四前列腺中部分割网络和第四前列腺底部分割网络,所述第四前列腺尖部分割网络用于分割前列腺MR图像得到第四前列腺尖部图像,所述第四前列腺中部分割网络用于分割前列腺MR图像得到第四前列腺中部图像,所述第四前列腺底部分割网络用于分割前列腺MR图像得到第四前列腺底部图像,将所述第四前列腺尖部图像、第四前列腺中部图像和第四前列腺底部图像进行合并,得到DCE图像前列腺全腺体分割图像。
[0060] 在一个具体实施例中,所述基于深度学习卷积神经网络的自动分割前列腺全腺体的系统,包括。
[0061] 获取单元,用于获取样本的前列腺MR图像;
[0062] 分类单元,用于将前列腺MR图像进行分类,将所述前列腺MR图像分为前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像;分割单元,用于将所述前列腺尖部MR图像、前列腺中部MR图像或前列腺底部MR图像分别输入前列腺尖部分割网络、前列腺中部分割网络和前列腺底部分割网络,得到前列腺尖部图像、前列腺中部图像和前列腺底部图像,将所述前列腺尖部图像、前列腺中部图像和前列腺底部图像进行图像合并,得到前列腺全腺体分割图像。
[0063] 图3是本发明实施例提供的一种基于深度学习卷积神经网络的自动分割前列腺全腺体的设备,包括:存储器和处理器;
[0064] 所述存储器用于存储程序指令;
[0065] 所述处理器用于调用程序指令,当程序指令被执行时实现上述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法步骤。
[0066] 本发明的一个目的在于提供一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现上述的基于深度学习卷积神经网络的自动分割前列腺全腺体方法步骤。
[0067] 本验证实施例的验证结果表明,为适应症分配固有权重相对于默认设置来说可以适度改善本方法的性能。
[0068] 所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的系统,装置和单元的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
[0069] 在本申请所提供的几个实施例中,应该理解到,所揭露的系统,装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
[0070] 所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
[0071] 另外,在本发明各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。
[0072] 本领域普通技术人员可以理解上述实施例的各种方法中的全部或部分步骤是可以通过程序来指令相关的硬件来完成,该程序可以存储于一计算机可读存储介质中,存储介质可以包括:只读存储器(ROM,Read Only Memory)、随机存取存储器(RAM,Random Access Memory)、磁盘或光盘等。
[0073] 本领域普通技术人员可以理解实现上述实施例方法中的全部或部分步骤是可以通过程序来指令相关的硬件完成,所述的程序可以存储于一种计算机可读存储介质中,上述提到的存储介质可以是只读存储器,磁盘或光盘等。
[0074] 以上对本发明所提供的一种计算机设备进行了详细介绍,对于本领域的一般技术人员,依据本发明实施例的思想,在具体实施方式及应用范围上均会有改变之处,综上所述,本说明书内容不应理解为对本发明的限制。