一种基于上下文信息的轻量级婴儿脑组织图像分割方法

文档序号:30183519发布日期:2022-05-26 16:08阅读:120来源:国知局
一种基于上下文信息的轻量级婴儿脑组织图像分割方法

1.本发明涉及图像处理技术领域,更具体的说是涉及一种基于上下文信息的轻量级婴儿脑组织图像分割方法。


背景技术:

2.近年来,人工智能和计算机视觉的技术正在迅速崛起,并且随着互联网的发展带来了巨量的数据用于实际应用,从海量的图像和视频数据中快速获取关键信息已成为计算机视觉领域的关键问题。在这其中,医学影像分析越来越受到人们的关注。利用核磁共振成像,可以收集到婴儿脑组织影像。利用这些数据结合计算机视觉技术,就可以在生命健康领域上大展身手。
3.理想的智能辅助诊疗需要对脑组织有正确的视觉理解。利用深度学习方法,婴儿脑组织分割智能辅助诊疗任务被定义为像素级语义分割问题,可以直接进行像素级别端到端(end-to-end)的语义理解。具体一些,即只需要在训练集中的影像与标签输入到模型框架中进行训练,得到权重与模型,然后在验证集中进行婴儿脑组织图像分割,通过不断调优和观察检测情况来得到最好的预测模型,最后利用预测模型对测试集进行测试,得到婴儿脑组织图像分割结果。基于深度学习的婴儿脑组织分割智能辅助诊疗方法,其核心是利用了卷积神经网络构建的多模态密集融合传递方法,它的强大之处在于它可以融合不同层级、不同模态下的特征,并且学习到极具鉴别性的特征,利用这些特征可以从多角度分割脑组织区域,并且自底向上密集传递来进行婴儿脑组织分割。卷积神经网络的架构主要包括两个阶段:编码阶段和解码阶段。编码阶段是指对输入影像进行编码,从而得到不同等级的编码特征;解码阶段是指对得到不同等级的编码特征进行解码组合,最后解码得到分割图。
4.随着各种分割模型的出现,最终的分割性能指标也随之不断上升,但是考虑到这些分割模型的存储量都比较大,运算浮点数都很大,要想部署在移动端就会有比较大的挑战。


技术实现要素:

5.有鉴于此,本发明提供了一种基于上下文信息的轻量级婴儿脑组织图像分割方法,在保证分割准确率的同时,降低其存储大小及其运算浮点数,构建一个多模态的轻量级分割网络。
6.为了实现上述目的,本发明提供如下技术方案:
7.一种基于上下文信息的轻量级婴儿脑组织图像分割方法,具体步骤包括:
8.选取原始婴儿脑组织的第一核磁共振影像、第二核磁共振影像和婴儿脑组织失真分割标注图构成训练集;
9.构建卷积神经网络,所述第一核磁共振影像和所述第二核磁共振影像依次进行特征提取采用多模态输入,并进行多尺度融合,将多尺度融合的结果经过相加、同时经多区域锐化进行堆叠;
10.将训练集中的每幅第一核磁共振影像及其对应的第二核磁共振影像作为原始输入影像,输入到所述卷积神经网络进行训练,得到训练集中的每幅核磁共振成像t1模态对应的分割预测图;
11.计算训练集中的每幅第一核磁共振影像对应的分割预测图构成的集合与对应的婴儿脑组织失真分割标注图构成的集合之间的损失函数值,采用二分类交叉熵损失函数获得,再通过adam调节器对网络参数进行更新;
12.重复训练并计算损失函数值,确定最优权值矢量和最优偏置项。
13.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,所述第一核磁共振影像,第二核磁共振影像和婴儿脑组织失真分割标注图为单通道图像。
14.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,所述卷积神经网络包括输入层、隐层和输出层;隐层包括依次设置的10个特征提取模块、10个多尺度融合模块、5个多区域锐化模块。
15.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,输入层的输入端接收第一核磁共振影像及其第二核磁共振影像,输入层的输出端输出原始输入第一核磁共振影像与第二核磁共振影像给隐层;输出层的输出端接收第5个多区域锐化模块中的所有特征图,输出层的输出端输出1幅与原始输入第一核磁共振影像对应的分割预测图。
16.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,所述多尺度融合模块包括:第1个多尺度融合模块、第2个多尺度融合模块、第3个多尺度融合模块、第4个多尺度融合模块、第5个多尺度融合模块、第6个多尺度融合模块、第7个多尺度融合模块、第8个多尺度融合模块、第9个多尺度融合模块、第10个多尺度融合模块;所述多尺度融合模块结构相同。
17.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,所述多尺度融合模块包括:第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第1个多尺度融合模块的输入端接收特征提取模块输出的五组特征图,每个尺寸调节模块经过不同倍上采样,经过所述第一个激活函数,与其中一组特征图相乘,再与所述特征图相加,输出多尺度融合后的特征图。
18.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,所述多区域锐化模块包括第1个多区域锐化模块、第2个多区域锐化模块、第3个多区域锐化模块、第4个多区域锐化模块、第5个多区域锐化模块;所述多区域锐化模块结构相同。
19.可选的,在上述的一种基于上下文信息的轻量级婴儿脑组织图像分割方法中,所述多区域锐化模块依次设置第一个空洞卷积模块、第二个空洞卷积模块、第一个池化模块、第一个激活函数;第1个多区域锐化模块的输入端接收第一核磁共振影像与第二核磁共振影像经过多尺度融合后的相加得到所有特征图,所有特征图通过所述第一个池化模块和第一个激活函数的输出结果,与所有特征图分别经过所述第一个空洞卷积模块、第二个空洞卷积模块堆叠的结果相乘,并与堆叠结果相加输出锐化后的特征图。
20.经由上述的技术方案可知,与现有技术相比,本发明公开提供了一种基于上下文信息的轻量级婴儿脑组织图像分割方法,在搭建网络时,采用的时轻量级的网络框架mobilenetv2以减少其网络参数量与内存占用并要设计出一些高效的特征融合、锐化模块
和更为有效的训练方式来进一步的提高其检测指标;采用多模态的输入,这意味着网络的输入包含多种不同种类的信息,核磁共振成像t1模态和核磁共振成像t2模态的影像信息,利用多尺度融合模块将不同信息有效利用,通过多区域锐化模块增加特征的边缘信息,可以更好分割图像;在现有的公布数据集上无论是分割速度还是分割效果都有很好的表现。
附图说明
21.为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据提供的附图获得其他的附图。
22.图1为本发明方法的总体实现框图;
23.图2为多尺度融合模块框图;
24.图3为多区域锐化模块框图。
具体实施方式
25.下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
26.本发明实施例公开了一种基于上下文信息的轻量级婴儿脑组织图像分割方法,保证分割准确率的同时,降低其存储大小及其运算浮点数。
27.本发明提出的一种基于上下文信息的轻量级婴儿脑组织图像分割方法,其总体实现框图如图1所示,其包括训练阶段和测试阶段两个过程;
28.所述的训练阶段过程的具体步骤为:
29.步骤1_1:选取q幅原始婴儿脑组织分割智能辅助诊疗核磁共振成像t1模态和核磁共振成像t2模态影像,并构成训练集,将训练集中的第q幅核磁共振成像t1模态记为将训练集中与对应的核磁共振成像t2模态和对应的婴儿脑组织真实分割标注图分别记为其中,核磁共振成像t1模态,核磁共振成像t2模态和婴儿脑组织分割影像为单通道图像,q为正整数,q≥200,如取q=250,q为正整数,1≤q≤q,1≤i≤w,1≤j≤h,w表示的宽度,h表示的高度,如取w=224、h=224,表示中坐标位置为(i,j)的像素点的像素值,表示中坐标位置为(i,j)的像素点的像素值,表示中坐标位置为(i,j)的像素点的像素值;在此,核磁共振成像t1模态,核磁共振成像t2模态和婴儿脑组织失真分割标注图直接选用公开的婴儿脑组织分割iseg2017数据集中核磁共振成像t1模态,核磁共振成像t2模态和婴儿脑组织失真分割标注图。
30.步骤1_2:构建卷积神经网络:卷积神经网络包括输入层、隐层和输出层;隐层包括依次设置的第1个特征提取模块、第2个特征提取模块、第3个特征提取模块、第4个特征提取
模块、第5个特征提取模块、第6个特征提取模块、第7个特征提取模块、第8个特征提取模块、第9个特征提取模块、第10个特征提取模块、第1个多尺度融合模块、第2个多尺度融合模块、第3个多尺度融合模块、第4个多尺度融合模块、第5个多尺度融合模块、第6个多尺度融合模块、第7个多尺度融合模块、第8个多尺度融合模块、第9个多尺度融合模块、第10个多尺度融合模块、第1个多区域锐化模块、第2个多区域锐化模块、第3个多区域锐化模块、第4个多区域锐化模块、第5个多区域锐化模块。
31.对于输入层,输入层的输入端接收核磁共振成像t1模态及其核磁共振成像t2模态,核磁共振成像t1模态与核磁共振成像t2模态都是单通道影像,输入层的输出端输出原始输入核磁共振成像t1模态与核磁共振成像t2模态给隐层;其中,要求输入层的输入端接收的原始输入核磁共振成像t1模态与核磁共振成像t2模态的宽度为w、高度为h。
32.对于第1个特征提取模块,其由现有开源网络mobilenetv2的第一卷积层、第二卷积层组成;第1个特征提取模块的输入端接收输入层的输出端输出的原始核磁共振成像t1模态,第1个特征提取模块的输出端输出16幅特征图,将16幅特征图构成的集合记为r1;r1中的每幅特征图的宽度为高度为
33.对于第2个特征提取模块,其由现有开源网络mobilenetv2的第三卷积层、第四卷积层组成;第2个特征提取模块的输入端接收r1中所有的特征图,第1个特征提取模块的输出端输出24幅特征图,将24幅特征图构成的集合记为r2;r2中的每幅特征图的宽度为高度为
34.对于第3个特征提取模块,其由现有开源网络mobilenetv2的第五卷积层、第六卷积层、第七卷积层组成;第3个特征提取模块的输入端接收r2中所有的特征图,第3个特征提取模块的输出端输出32幅特征图,将32幅特征图构成的集合记为r3;r3中的每幅特征图的宽度为高度为
35.对于第4个特征提取模块,其由现有开源网络mobilenetv2的第八卷积层、第九卷积层、第十卷积层、第十一卷积层、第十二卷积层、第十三卷积层、第十四卷积层、第十五卷积层、第十六卷积层、第十七卷积层组成;第4个特征提取模块的输入端接收r3中所有的特征图,第4个特征提取模块的输出端输出160幅特征图,将160幅特征图构成的集合记为r4;r4中的每幅特征图的宽度为高度为
36.对于第5个特征提取模块,其由现有开源网络mobilenetv2的第十八卷积层组成;第5个特征提取模块的输入端接收r4中所有的特征图,第5个特征提取模块的输出端输出320幅特征图,将320幅特征图构成的集合记为r5;r5中的每幅特征图的宽度为高度为
37.对于第6个特征提取模块,其由现有开源网络mobilenetv2的第一卷积层、第二卷积层组成;第6个特征提取模块的输入端接收输入层的输出端输出的原始核磁共振成像t2模态,第6个特征提取模块的输出端输出16幅特征图,将16幅特征图构成的集合记为r6;r6中
的每幅特征图的宽度为高度为
38.对于第7个特征提取模块,其由现有开源网络mobilenetv2的第三卷积层、第四卷积层组成;第7个特征提取模块的输入端接收r6中所有的特征图,第6个特征提取模块的输出端输出24幅特征图,将24幅特征图构成的集合记为r7;r7中的每幅特征图的宽度为高度为
39.对于第8个特征提取模块,其由现有开源网络mobilenetv2的第五卷积层、第六卷积层、第七卷积层组成;第8个特征提取模块的输入端接收r7中所有的特征图,第8个特征提取模块的输出端输出32幅特征图,将32幅特征图构成的集合记为r8;r8中的每幅特征图的宽度为高度为
40.对于第9个特征提取模块,其由现有开源网络mobilenetv2的第八卷积层、第九卷积层、第十卷积层、第十一卷积层、第十二卷积层、第十三卷积层、第十四卷积层、第十五卷积层、第十六卷积层、第十七卷积层组成;第9个特征提取模块的输入端接收r8中所有的特征图,第9个特征提取模块的输出端输出160幅特征图,将160幅特征图构成的集合记为r9;r9中的每幅特征图的宽度为高度为
41.对于第10个特征提取模块,其由现有开源网络mobilenetv2的第十八卷积层组成;第10个特征提取模块的输入端接收r9中所有的特征图,第10个特征提取模块的输出端输出320幅特征图,将320幅特征图构成的集合记为r
10
;r
10
中的每幅特征图的宽度为高度为
42.对于第1个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第1个多尺度融合模块的输入端(从左到右)依次接收r1、r2、r3、r4、r5中的所有特征图,第1个多尺度融合模块的输出端输出16幅特征图,将16幅特征图构成的集合记为f1;其中第一个尺寸调节模块采用的是2倍上采样(upsample),第二个尺寸调节模块采用的是4倍上采样,第三个尺寸调节模块采用的是8倍上采样,第四个尺寸调节模块采用的16倍上采样,第一个激活函数方式为“relu”,f1中的每幅特征图的宽度为高度为
43.对于第2个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第2个多尺度融合模块的输入端依次接收r1、r2、r3、r4、r5中的所有特征图,第2个多尺度融合模块的输出端输出24幅特征图,将24幅特征图构成的集合记为f2;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是2倍上采样(upsample),第三个尺寸调节模块采用的是4倍上采样,第四个尺寸调节模块采用的8倍上采样,第一个激活函数方式为“relu”,f2中的每幅特征图的宽度为高度为
44.对于第3个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第3个多尺度融合模块的输入端依次接收r1、r2、r3、r4、r5中的所有特征图,第3个多尺度融合模块的输出端输出32幅特征图,将32幅特征图构成的集合记为f3;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是两次最大池化下采样(maxpool),第三个尺寸调节模块采用的是2倍上采样(upsample),第四个尺寸调节模块采用的4倍上采样,第一个激活函数方式为“relu”,f3中的每幅特征图的宽度为高度为
45.对于第4个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第4个多尺度融合模块的输入端依次接收r1、r2、r3、r4、r5中的所有特征图,第4个多尺度融合模块的输出端输出160幅特征图,将160幅特征图构成的集合记为f4;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是两次最大池化下采样,第三个尺寸调节模块采用的是三次最大池化下采样,第四个尺寸调节模块采用的2倍上采样(upsample),第一个激活函数方式为“relu”,f4中的每幅特征图的宽度为高度为
46.对于第5个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第5个多尺度融合模块的输入端依次接收r1、r2、r3、r4、r5中的所有特征图,第5个多尺度融合模块的输出端输出320幅特征图,将320幅特征图构成的集合记为f5;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是两次最大池化下采样,第三个尺寸调节模块采用的是三次最大池化下采样,第四个尺寸调节模块采用的是三次最大池化下采样,第一个激活函数方式为“relu”,f5中的每幅特征图的宽度为高度为
47.对于第6个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第6个多尺度融合模块的输入端(从左到右)依次接收r6、r7、r8、r9、r
10
中的所有特征图,第6个多尺度融合模块的输出端输出16幅特征图,将16幅特征图构成的集合记为f6;其中第一个尺寸调节模块采用的是2倍上采样(upsample),第二个尺寸调节模块采用的是4倍上采样,第三个尺寸调节模块采用的是8倍上采样,第四个尺寸调节模块采用的16倍上采样,第一个激活函数方式为“relu”,f6中的每幅特征图的宽度为高度为
48.对于第7个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第7个多尺度融合模块的输入端依次接收r6、r7、r8、r9、r
10
中的所有特征图,第7个多尺度融合模块的输出端输出24幅特征图,将24幅特征图构成的集合记为f7;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是2倍上采样(upsample),第三个尺寸调节模块采用的是4倍上采样,第四个尺寸调节模块采用的8倍上采样,第一个激活函数方式为“relu”,f7中的每幅特征图的宽度为高度为
49.对于第8个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第8个多尺度融合模块的输入端依次接收r6、r7、r8、r9、r
10
中的所有特征图,第8个多尺度融合模块的输出端输出32幅特征图,将32幅特征图构成的集合记为f8;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是两次最大池化下采样(maxpool),第三个尺寸调节模块采用的是2倍上采样(upsample),第四个尺寸调节模块采用的4倍上采样,第一个激活函数方式为“relu”,f8中的每幅特征图的宽度为高度为
50.对于第9个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第9个多尺度融合模块的输入端依次接收r6、r7、r8、r9、r
10
中的所有特征图,第9个多尺度融合模块的输出端输出160幅特征图,将160幅特征图构成的集合记为f9;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是两次最大池化下采样,第三个尺寸调节模块采用的是三次最大池化下采样,第四个尺寸调节模块采用的2倍上采样(upsample),第一个激活函数方式为“relu”,f9中的每幅特征图的宽度为高度为
51.对于第10个多尺度融合模块,其由依次设置的第一个尺寸调节模块、第二个尺寸调节模块、第三个尺寸调节模块、第四个尺寸调节模块和第一个激活函数组成;第10个多尺度融合模块的输入端依次接收r6、r7、r8、r9、r
10
中的所有特征图,第10个多尺度融合模块的输出端输出320幅特征图,将320幅特征图构成的集合记为f
10
;其中第一个尺寸调节模块采用的是一次最大池化下采样(maxpool),第二个尺寸调节模块采用的是两次最大池化下采样,第三个尺寸调节模块采用的是三次最大池化下采样,第四个尺寸调节模块采用的是三次最大池化下采样,第一个激活函数方式为“relu”,f
10
中的每幅特征图的宽度为高度为
52.对于第1个多区域锐化模块,其由依次设置的第一个空洞卷积模块、第二个空洞卷积模块、第一个池化模块、第一个激活函数组成;第1个多区域锐化模块的输入端接收f5和f
10
直接相加(element-wise addition)后的所有特征图,第1个多区域锐化模块的输出端输出320幅特征图,将320幅特征图构成的集合记为o5;其中第一个空洞卷积模块的卷积核大小(kernel-size)为3
×
3、步距(stride)为1、填充(padding)为1、空洞率(dilation)为1,第二个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为7、空洞率为7,第一个池化模块采用的是自适应平均池化,第一个激活函数的激活方式为“sigmoid”,o5中的每幅特征图的宽度为高度为
53.对于第2个多区域锐化模块,其由依次设置的第一个空洞卷积模块、第二个空洞卷积模块、第一个池化模块、第一个激活函数组成;第2个多区域锐化模块的输入端接收f4和f9直接相加后的特征图再与o5堆叠(concatenate)后的所有特征图,第2个多区域锐化模块的输出端输出160幅特征图,将160幅特征图构成的集合记为o4;其中第一个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为1、空洞率为1,第二个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为7、空洞率为7,第一个池化模块采用的是自适应平均池化,第一个激活函数的激活方式为“sigmoid”,o4中的每幅特征图的宽度为高度为
54.对于第3个多区域锐化模块,其由依次设置的第一个空洞卷积模块、第二个空洞卷积模块、第一个池化模块、第一个激活函数组成;第3个多区域锐化模块的输入端接收f3和f8直接相加后的特征图再与o4和o5堆叠后的所有特征图,第3个多区域锐化模块的输出端输出32幅特征图,将32幅特征图构成的集合记为o3;其中第一个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为1、空洞率为1,第二个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为7、空洞率为7,第一个池化模块采用的是自适应平均池化,第一个激活函数的激活方式为“sigmoid”,o3中的每幅特征图的宽度为高度为
55.对于第4个多区域锐化模块,其由依次设置的第一个空洞卷积模块、第二个空洞卷积模块、第一个池化模块、第一个激活函数组成;第4个多区域锐化模块的输入端接收f2和f7直接相加后的特征图再与o3和o4堆叠后的所有特征图,第4个多区域锐化模块的输出端输出24幅特征图,将24幅特征图构成的集合记为o2;其中第一个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为1、空洞率为1,第二个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为7、空洞率为7,第一个池化模块采用的是自适应平均池化,第一个激活函数的激活方式为“sigmoid”,o2中的每幅特征图的宽度为高度为
56.对于第5个多区域锐化模块,其由依次设置的第一个空洞卷积模块、第二个空洞卷积模块、第一个池化模块、第一个激活函数组成;第5个多区域锐化模块的输入端接收f1和f6直接相加后的特征图再与o2和o3堆叠后的所有特征图,第5个多区域锐化模块的输出端输出16幅特征图,将16幅特征图构成的集合记为o1;其中第一个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为1、空洞率为1,第二个空洞卷积模块的卷积核大小为3
×
3、步距为1、填充为7、空洞率为7,第一个池化模块采用的是自适应平均池化,第一个激活函数的激活方式为“sigmoid”,o1中的每幅特征图的宽度为高度为
57.对于输出层,其由第一卷积层和双倍上采样组成,其中第一卷积层的卷积核大小为3
×
3、步距为1、填充为1、空洞率为1,双倍上采样采用的双线性插值上采样;输出层的输出端接收o1中的所有特征图,输出层的输出端输出1幅与原始输入核磁共振成像t1模态对应的分割预测图。
58.步骤1_3:将训练集中的每幅原始核磁共振成像t1模态及其对应的核磁共振成像t2模态作为原始输入影像,输入到我们所设计的网络中进行训练,得到训练集中的每幅核磁共振成像t1模态对应的1幅分割预测图,将对应的分割预测图构成的集合记为
59.步骤1_4:计算训练集中的每幅核磁共振成像t1模态对应的分割预测图构成的集合与对应的婴儿脑组织失真分割标注图构成的集合之间的损失函数值,将与之间的损失函数值记为之间的损失函数值记为采用二分类交叉
熵损失函数获得,再通过adam调节器对网络参数进行更新。
60.步骤1_5:重复执行步骤1_3和步骤1_4共v次,得到卷积神经网络婴儿脑组织分割智能辅助诊疗训练模型,在本实施例中取v=100。
61.所述的测试阶段过程的具体步骤为:
62.步骤2_1:令{kr(i,j)}、{k
t
(i,j)}表示待婴儿脑组织分割智能辅助诊疗的核磁共振成像t1模态及其对应的核磁共振成像t2模态;其中,1≤i≤w,1≤j≤h,w表示{kr(i,j)}和{k
t
(i,j)}的宽度,h表示{kr(i,j)}和{k
t
(i,j)}的高度,kr(i,j)}表示{kr(i,j)}中坐标位置为(i,j)的像素点的像素值,k
t
(i,j)}表示{k
t
(i,j)}中坐标位置为(i,j)的像素点的像素值。
63.步骤2_2:将{kr(i,j)}和{k
t
(i,j)}的中的核磁共振成像t1模态及其对应的核磁共振成像t2模态输入到卷积神经网络训练模型中,得到核磁共振成像t1模态kr(i,j)对应的分割预测图像k
pre
(i,j),将这些预测图像组成的集合记为{k
pre
(i,j)}。
64.为了进一步验证本发明方法的可行性和有效性,进行实验。
65.使用基于python语言的pytorch的深度学习库搭建卷积神经网络的架构。采用iseg2017数据集进行训练。为了验证该神经网络架构的有效性,本文的评价指标使用了dsc(dice similarity coeffificient)和asd(average surface distance),实验结果如表1所示。从表1所列的数据可知,按本发明方法得到的婴儿脑组织分割结果是较好的,表明利用本发明方法来获取婴儿脑组织分割智能辅助诊疗是可行性且有效的。
66.网络权重(size)、网络参数量(parameters)、每秒传输帧数(frames per second,fps)来评价所设计的神经网络轻量级性能,实验结果如表2所示。从表2所列的数据可知,本发明方法的推理速度及内存占用都有明显的优势,对于将来的移动端部署有着更大的优势。
67.表1 利用本发明方法在测试集上的评测结果
[0068][0069]
表2 本发明方法的各项其它指标
[0070][0071][0072]
本说明书中各个实施例采用递进的方式描述,每个实施例重点说明的都是与其他实施例的不同之处,各个实施例之间相同相似部分互相参见即可。对于实施例公开的装置而言,由于其与实施例公开的方法相对应,所以描述的比较简单,相关之处参见方法部分说明即可。
[0073]
对所公开的实施例的上述说明,使本领域专业技术人员能够实现或使用本发明。
对这些实施例的多种修改对本领域的专业技术人员来说将是显而易见的,本文中所定义的一般原理可以在不脱离本发明的精神或范围的情况下,在其它实施例中实现。因此,本发明将不会被限制于本文所示的这些实施例,而是要符合与本文所公开的原理和新颖特点相一致的最宽的范围。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1