一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

一种全类型主动脉夹层真假腔图像分割方法及系统与流程

2021-11-22 14:16:00 来源:中国专利 TAG:


1.本发明涉及图像分割技术领域,特别是涉及一种全类型主动脉夹层真假腔图像分割方法及系统。


背景技术:

2.现有的端到端的主动脉夹层真假腔图像分割方法以一系列ct图像组成的三维体数据作为输入,通过各种语义分割网络,实现主动脉夹层真假腔的分割。主动脉几何结构是圆形管腔,但由于其在人体内呈现弯曲状态,其ct图像序列并不完全是规则的圆形,给分割带来难度,故非端到端的主动脉夹层分割方法增加了提取主动脉中心线这一步骤,通过沿中心线的多平面重建,获取拉直后的主动脉三维图像。这一类研究工作以多平面重建后的主动脉夹层数据作为输入,恢复了主动脉圆形管腔的几何形态,使夹层具有更明显的特征,往往分割精度更高,但是其分割过程非全自动化。
3.因此,本领域亟需一种全自动化的高精度的全类型主动脉夹层真假腔图像分割的技术方案。


技术实现要素:

4.本发明的目的是提供一种全类型主动脉夹层真假腔图像分割方法及系统,以解决目前分割精度不高或者分割过程非全自动化的问题。
5.为实现上述目的,本发明提供了如下方案:
6.一种全类型主动脉夹层真假腔图像分割方法,所述方法包括:
7.获取待分割真假腔图像;
8.利用基于注意力机制的分割网络模型对所述待分割真假腔图像进行分割,得到分割结果掩膜图;
9.将所述分割结果掩膜图映射到所述待分割真假腔图像中,得到分割结果。
10.可选的,在所述利用基于注意力机制的分割网络模型对所述待分割真假腔图像进行分割,得到分割结果掩膜图之前,还包括:
11.根据主动脉区域ct值范围,对所述待分割真假腔图像进行阈值化处理,得到待分割阈值化数据;
12.利用灰度变换处理所述待分割阈值化数据,得到待分割灰度变换数据;灰度变换的公式为:ct
out_log
=10log(1 ct
out
),其中,ct
out
表示图像阈值处理后的ct值,ct
out_log
表示经过对数灰度变换后的ct值。
13.可选的,所述基于注意力机制的分割网络模型的训练方法包括:
14.获取真假腔图像;
15.对所述真假腔图像中的主动脉区域及真腔和假腔进行标注,得到真假腔数据;所述真假腔数据包括真假腔图像和标注之后的真假腔图像的掩膜图;
16.设定分割神经网络的超参数,利用3d卷积提取真假腔数据的主动脉夹层空间相关
性,并利用混合注意力模块作为基本单元进行上下采样对所述真假腔数据施加权重,通过指数函数归一化完成每一像素值类别的预测,得到具有类别信息的真假腔分割掩膜图;计算所述真假腔分割掩膜图与所述标注之后的真假腔图像的掩膜图之间的损失函数,直至结果收敛或者达到训练次数,得到分割网络模型。
17.可选的,在所述对所述真假腔图像中的主动脉区域及真腔和假腔进行标注,得到真假腔数据之后,还包括:
18.使用最近邻插值算法将所述真假腔数据的数据维度归一化,得到归一化真假腔数据;
19.对所述归一化真假腔数据进行裁剪,保留所述真假腔数据中的中心区域。
20.可选的,在所述对所述真假腔图像中的主动脉区域及真腔和假腔进行标注,得到真假腔数据之后,还包括:
21.根据主动脉区域ct值范围,对所述真假腔数据进行阈值化处理,得到阈值化数据;
22.利用灰度变换处理所述阈值化数据,得到灰度变换数据;灰度变换的公式为:ct
out_log
=10log(1 ct
out
),其中,ct
out
表示图像阈值处理后的ct值,ct
out_log
表示经过对数灰度变换后的ct值。
23.可选的,在所述利用灰度变换处理所述阈值化数据,得到灰度变换数据之后,还包括:
24.利用数值归一化处理公式,对所述灰度变换数据进行数值归一化处理;所述归一化处理公式为:其中,image
out
为经过数值归一化处理后的图像,image
in
为灰度变换处理后的图像,min_bound为图像中最小ct值,取0,max_bound为图像中最大ct值,取为800。
25.可选的,所述损失函数为:
[0026][0027]
其中,n
c
表示类别数,类别包括真腔、假腔和背景;i表示第i个类别;式中m
isb
为类间三维共享面度量矩阵,大小为n
c
×
n
c
,代表真假腔分割掩膜图的类间三维共享面度量矩阵,代表标注之后的真假腔图像的掩膜图的类间三维共享面度量矩阵。
[0028]
可选的,在得到分割网络模型之后,还包括:
[0029]
利用dice系数判断所述分割网络模型的性能是否达到要求;
[0030]
若各类别分割的dice系数均大于90%,则所述分割网络模型满足性能要求;
[0031]
若所述分割网络模型达不到性能要求,则调整所述超参数,重新训练网络模型。
[0032]
一种全类型主动脉夹层真假腔图像分割系统,所述系统包括:
[0033]
图像获取单元,用于获取待分割真假腔图像;
[0034]
图像分割单元,用于利用基于注意力机制的分割网络模型对所述待分割真假腔图
像进行分割,得到分割结果掩膜图;
[0035]
分割结果获取单元,用于将所述分割结果掩膜图映射到所述待分割真假腔图像中,得到分割结果。
[0036]
可选的,所述系统还包括:
[0037]
训练图像获取单元,用于获取真假腔图像;
[0038]
训练图像标注单元,用于对所述真假腔图像中的主动脉区域及真腔和假腔进行标注,得到真假腔数据;所述真假腔数据包括真假腔图像和标注之后的真假腔图像的掩膜图;
[0039]
分割网络模型训练单元,用于:
[0040]
设定分割神经网络的超参数,利用3d卷积提取真假腔数据的主动脉夹层空间相关性,并利用混合注意力模块作为基本单元进行上下采样对所述真假腔数据施加权重,通过指数函数归一化完成每一像素值类别的预测,得到具有类别信息的真假腔分割掩膜图;计算所述真假腔分割掩膜图与所述标注之后的真假腔图像的掩膜图之间的损失函数,直至结果收敛或者达到训练次数,得到分割网络模型。
[0041]
根据本发明提供的具体实施例,本发明公开了以下技术效果:
[0042]
本发明提供的全自动化的全类型主动脉夹层真假腔图像分割方法及系统,在保证分割精度的前提下解决了现有技术中的非全自动化问题。并且,针对主动脉夹层真假腔具有空间相关性的特点,提出使用基于注意力机制的主动脉夹层真假腔分割神经网络模型,使模型对于真假腔的区别更加敏感,从而提高了模型的精度。
附图说明
[0043]
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
[0044]
图1为本发明实施例一提供的全类型主动脉夹层真假腔图像分割方法流程图。
[0045]
图2为本发明实施例一提供的上下采样过程的基本单元混合注意力模块示意图。
[0046]
图3为本发明实施例一提供的3d卷积过程示意图
[0047]
图4为本发明实施例一提供的主动脉夹层真假腔图像分割网络示意图。
[0048]
图5为本发明实施例一提供的主动脉夹层真假腔图像分割系统的框图。
具体实施方式
[0049]
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
[0050]
目前为止,主动脉夹层真假腔图像分割方法主要有传统的方法,和目前最为流行的基于深度学习的方法两种。
[0051]
传统的主动脉夹层真假腔图像分割技术有基于模型的方法、基于海森矩阵的方法和基于小波分析的方法等,这些传统的方法利用主动脉形态的先验知识,结合传统的图像
处理方法,通过完整主动脉分割、内膜片检测和真假腔分离等多个步骤完成主动脉夹层的分割。但是存在以下局限性:1)强烈依赖于主动脉夹层先验知识,对于主动脉夹层动脉瘤、血栓等特殊情况,传统方法无法将真假腔分割出来。2)主动脉同时含有竖直和弯曲的形态,传统方法难以适用全段。3)多步骤实现主动脉夹层真假腔的分割,某一步骤失败即无法得到最后分割的结果。
[0052]
随着深度学习在计算机视觉领域快速的发展,在医学图像领域也有越来越多的相关研究,并取得不错的效果,在主动脉夹层真假腔图像的分割这一课题研究上,主要有端到端和非端到端的两种方法,各种研究工作都建立了各自非公开的主动脉夹层真假腔图像数据集。而主动脉夹层的标注是一项十分费时费力的工作,往往还需要结合专业知识,各数据集的准确性有待考究,这也是深度学习还未在主动脉夹层分割问题上深度开展的原因。
[0053]
端到端的主动脉夹层分割方法以一系列ct图像组成的三维体数据作为输入,通过各种语义分割网络,实现主动脉夹层真假腔的分割。主动脉几何结构是圆形管腔,但由于在人体内呈现弯曲状态,其ct图像序列并不完全是规则的圆形,给分割带来难度,故非端到端的主动脉夹层图像分割方法增加了提取主动脉中心线这一步骤,通过沿中心线的多平面重建,获取拉直后的主动脉三维图像。这一类研究工作以多平面重建后的主动脉夹层数据作为输入,恢复了主动脉圆形管腔的几何形态,使夹层具有更明显的特征,往往分割精度更高,但是其分割过程非全自动化。
[0054]
通过以上关于主动脉夹层真假腔图像分割的相关研究,可知目前的主动脉夹层真假腔图像分割工作存在以下问题:1)选择的数据过于理想,去除了含血栓等特殊情况。目前还没有公开的主动脉夹层的数据集,所有的工作都基于各自所建立的数据集,通过他们的实验结果发现,对于血栓、壁间血肿等属于主动脉夹层假腔的情况,未正确标注或未标注。2)未充分考虑图像中主动脉真假腔与其他组织的空间相关性。3)现有的主动脉夹层真假腔图像分割未考虑包括分支血管在内的完整的主动脉,目前的研究工作均没有对分支血管夹层情况进行分割。
[0055]
在此背景下,本发明提出了一种基于注意力机制的全类型主动脉夹层真假腔图像分割方法,实现了包括分支血管在内的主动脉a型和b型夹层的真假腔分割,不仅充分考虑了主动脉夹层的空间相关性,还通过注意力机制使便于真假腔区分的重要区域权重提升,使模型精度进一步提升。另外通过引入类间三维共享面损失函数,解决了某些a型样本中真假腔类别不平衡从而模型真假腔分割精度低的问题,实现了a型和b型主动脉夹层真假腔的分割。
[0056]
本发明的目的是提供一种全类型主动脉夹层真假腔图像分割方法及系统,以解决目前分割精度不高或者分割过程非全自动化的问题
[0057]
为使本发明的上述目的、特征和优点能够更加明显易懂,下面结合附图和具体实施方式对本发明作进一步详细的说明。
[0058]
实施例一:
[0059]
如图1所示,本发明实施例提供了一种全类型主动脉夹层真假腔图像分割方法,该方法包括:
[0060]
s1、获取待分割真假腔图像;
[0061]
获取到待分割真假腔图像之后,先要对其进行预处理过程,具体包括:
[0062]
(1)数据维度归一化
[0063]
一方面网络模型的输入需要相同数据维度,另一方面大数据训练对计算机显存有较大的需求。因此第一步我们需要将数据维度归一化,使用最近邻插值算法,ndimage库中zoom函数可实现这一功能。接下来,为减少其他组织区域的影响,裁剪包含主动脉的中心区域。
[0064]
(2)数值归一化及灰度变换
[0065]
其次,增强ct图像的ct值可以反应人体组织或器官密度大小,计量单位为hu,不同组织ct值不同,根据主动脉区域ct值范围,对所述待分割真假腔图像进行阈值化处理,得到待分割阈值化数据,对裁剪后的待分割真假腔图像进行阈值化处理,可以进一步排除背景区域的干扰,具体处理为:
[0066][0067]
其中,ct
in
表示原始待分割真假腔图像的ct值,ct
out
表示图像阈值处理后待分割真假腔图像的ct值,经过阈值化处理,ct值小于0的区域全部被抑制。由于部分主动脉夹层假腔中造影剂的流动速度慢或不含造影剂,故在待分割真假腔图像的ct图像中ct值与其他组织较接近,特别在边界处肉眼较难区分开,本发明使用灰度变换对待分割阈值化数据进行处理,以增大低ct值区域的差别,使之边界更为明显,具体使用对数变换,使变换后的ct值差异增大,得到待分割灰度变换数据:
[0068]
ct
out_log
=10log(1 ct
out
)
[0069]
其中,ct
out
表示待分割阈值化数据的ct值,ct
out_log
表示待分割灰度变换数据的ct值。
[0070]
最后,由于灰度变换后得到的待分割灰度变换数据中主动脉区域ct值差别仍然很大,在0

800之间,会导致后续神经网络模型训练不稳定,因此需要通过下式对待分割灰度变换数据中的ct值进行归一化处理:
[0071][0072]
其中,image
out
为经过数值归一化后的图像,image
in
为待分割灰度变换数据,min_bound为图像中最小ct值,取0,max_bound为图像中最大ct值,取为800,经过此归一化处理,数据集图像像素值在0

1之间。
[0073]
s2、利用基于注意力机制的分割网络模型对所述待分割真假腔图像进行分割,得到分割结果掩膜图;
[0074]
s3、将所述分割结果掩膜图映射到所述待分割真假腔图像中,得到分割结果。
[0075]
具体的,本发明实施例中所述基于注意力机制的分割网络模型的训练方法包括:
[0076]
a1、获取真假腔图像;
[0077]
构建包含分支血管在内的全类型主动脉夹层真假腔图像数据集,原始数据为来源于某医院影像科200例主动脉夹层患者的胸腹cta图像,按照dicom协议存储,每一例数据由数百至上千张不等的图像组成,一张图像对应一个.dcm文件,除了影像外,一例数据中还包括了ct扫描机器信息、扫描层厚、时间信息和患者信息等。
[0078]
a2、对所述真假腔图像中的主动脉区域及真腔和假腔进行标注,得到真假腔数据;
所述真假腔数据包括真假腔图像和标注之后的真假腔图像的掩膜图;
[0079]
使用软件3d slicer可打开原始图像,原始图像大小为512
×
512
×
s(s为图像张数,500

2000张不等),为训练分割网络模型,需要对每一例数据每一张图像中主动脉区域及真腔和假腔进行人工标注,标注范围由上至下,涉及从头臂干动脉、左颈总动脉和左锁骨下动脉到股动脉,并包含冠状动脉、肾动脉、肠系膜上、下动脉等分支血管的主动脉区域。经过标注并保存后,得到由200例数据组成的数据集,按照6:2:2的比例可以分为训练集、验证集和测试集,每一例数据包含.nii格式的真假腔图像的原始图像和对应的标注了完整主动脉和真假腔的真假腔图像的掩膜图,真假腔图像的掩膜图为.nrrd格式,此人工标注结果即作为真实值用于后续步骤中网络训练的标准。
[0080]
经标注后形成的数据集,要用于神经网络模型训练还需要数据预处理,包括数据维度归一化、数据灰度变换和数值归一化三个方面。
[0081]
(1)数据维度归一化
[0082]
首先,数据集中的数据大小为512
×
512
×
s,一方面网络模型的输入需要相同数据维度,另一方面大数据训练对计算机显存有较大的需求。因此第一步需要将数据维度归一化,使用最近邻插值算法将所述真假腔数据的数据维度归一化,得到归一化真假腔数据,ndimage库中zoom函数可实现这一功能,归一化后数据维度由512
×
512
×
s变成统一的256
×
256
×
256。接下来,为减少其他组织区域的影响,对所述归一化真假腔数据进行裁剪,保留所述真假腔数据中的中心区域,数据维度变为128
×
128
×
256。
[0083]
(2)数值归一化及灰度变换
[0084]
增强ct图像的ct值可以反应人体组织或器官密度大小,计量单位为hu,不同组织ct值不同,根据主动脉区域ct值范围,对裁剪后的图像进行阈值化处理,可以进一步排除背景区域的干扰,得到阈值化数据,具体处理为:
[0085][0086]
其中,ct
in
表示裁剪后的图像的ct值,ct
out
表示图像阈值处理后的ct值,经过阈值化处理,ct值小于0的区域全部被抑制。由于部分主动脉夹层假腔中造影剂的流动速度慢或不含造影剂,故在cta图像中ct值与其他组织较接近,特别在边界处肉眼较难区分开,本发明使用灰度变换对阈值化数据进行处理,以增大低ct值区域的差别,使之边界更为明显,具体使用对数变换,使变换后的ct值差异增大:
[0087]
灰度变换的公式为:ct
out_log
=10log(1 ct
out
),其中,ct
out
表示阈值化数据的ct值,ct
out_log
表示经过对数灰度变换后的ct值。
[0088]
最后,由于灰度变换后阈值化数据中主动脉区域ct值差别仍然很大,在0

800之间,会导致后续神经网络模型训练不稳定,因此需要利用数值归一化处理公式,对所述灰度变换数据进行数值归一化处理;所述归一化处理公式为:其中,image
out
为经过数值归一化处理后的图像,image
in
为灰度变换处理后的阈值化数据,min_bound为图像中最小ct值,取0,max_bound为图像中最大ct值,取为800,经过此归一化处理,数据集图像像素值在0

1之间,大小均为128
×
128
×
256。
[0089]
a3、设定分割神经网络的超参数,利用3d卷积提取真假腔数据的主动脉夹层空间相关性,并利用混合注意力模块作为基本单元进行上下采样对所述真假腔数据施加权重,通过指数函数归一化完成每一像素值类别的预测,得到具有类别信息的真假腔分割掩膜图;计算所述真假腔分割掩膜图与所述标注之后的真假腔图像的掩膜图之间的损失函数,直至结果收敛或者达到训练次数,得到分割网络模型。
[0090]
所述损失函数为:
[0091][0092]
其中,n
c
表示类别数,类别包括真腔、假腔和背景;i表示第i个类别;式中m
isb
为类间三维共享面度量矩阵,大小为n
c
×
n
c
,代表真假腔分割掩膜图的类间三维共享面度量矩阵,代表标注之后的真假腔图像的掩膜图的类间三维共享面度量矩阵。
[0093]
作为一种具体的实施方式,本实施例中利用如下方法设计训练分割网络模型:
[0094]
首先需要设定主动脉夹层真假腔分割神经网络的超参数,包括学习速率(learning rate=0.0005),迭代次数(epoch=100),批处理数量(batch size=4)等,之后可以开始训练网络。此处所使用的数据集为训练集和验证集。
[0095]
该步骤主要是设计主动脉夹层真假腔分割的神经网络模型,所提出的方法基础结构与attention

unet网络相似,但基础单元不同。下面分别介绍各个单元:
[0096]
(1)基本单元混合注意力模块
[0097]
如图2所示为上下采样过程的基本单元混合注意力模块,输入为通道数为n的特征图,此处的特征图即为上述经过数据预处理之后的图像,经过第一步3d conv bn relu三层操作得到初步提取特征的通道数为2n的特征图。3dconv即3d卷积,如图3所示,通过3*3*3的卷积核过滤器依次在特征图上遍历计算,即全特征图共享卷积核参数,可以提取到特征图上的信息。使用3d卷积,适用于主动脉夹层三维数据,进而掌握主动脉夹层空间相关性。bn指归一化层,对输入的特征图进行减均值除以方差的操作,作用一是加快了模型的训练速度,作用二则是限制特征图数据的分布,有利于模型训练。最后relu层为激活层,通过如下式所示的relu函数对特征图进行非线性映射,以增加模型非线性表达能力。
[0098]
f(x)=max(0,x)
[0099]
第二步仍然是3d卷积,进一步提取特征图的抽象信息。第三步为3ddilation

conv,即3d空洞卷积,卷积过程和普通的3d卷积相同,不同的地方在于卷积核过滤器数值之间有间隔,可以在保持卷积核参数大小不变的同时,增大卷积核的视野。经过第四步再一次3d卷积后,通过如下式所示的sigmoid激活函数层,将特征图数值映射到[0,1]之间,至此得到了原输入特征图的权重图,图2中权重图内不同花色表示不同权重。
[0100][0101]
将此权重图与第二步所得的特征图相乘后,即给特征图赋予了权重,得到加权的
特征图,最后与第一步得到的特征图相加,即结合了加权值与特征值,得到最后输出特征图。通过神经网络迭代训练计算上述权重,真假腔区域的权重将上升,模型更能关注到决定主动脉真假腔的关键部位,加强了模型对于空间相关性的敏感度,增加模型分割准确度。
[0102]
(2)整体网络模型
[0103]
该网络的输入为步骤a2中所得到的image
out
,尺寸为128
×
128
×
256,网络最终层为softmax层,通过指数函数归一化来完成每一像素值类别的预测,输出为尺寸3
×
128
×
128
×
256的掩膜图,3代表了类别数,即代表输出值有三个维度,取值为0,1,2,维度0代表背景,维度1代表真腔,维度2代表假腔,在每一维度和原图对应的128
×
128
×
256大小体数据中,像素值为0和1,1即代表此像素被预测为该维度对应的类别。从数据预处理到网络输出整体流程的主动脉真假腔图像分割网络如图4所示。
[0104]
整体网络为一个u型结构,基本单元即上述混合注意力模块,网络主要可以分为下采样、上采样、跳跃连接三个部分,下采样通过最大值池化实现,经过一次最大值池化,通道数增加,特征图尺寸减小,即逐步获取特征图抽象信息。上采样过程通过反卷积实现,经过一次反卷积,通道数减小,特征图尺寸增加,会恢复到原输入特征图大小。跳跃连接使用了注意力门控模块,其获取权重的过程类似于上述混合注意力模块,通过卷积加上激活函数得到特征图的权重,将所得的权重乘以下一层特征图即给特征图赋予了权重,最后得到的特征图与下一层特征图上采样后的特征图拼接在一起形成改成混合注意力模块的输入。
[0105]
经过主动脉真假腔图像分割网络,即为一次前向传播,得到主动脉夹层真假腔的预测值,此处的预测值即为具有类别信息的真假腔分割掩膜图,神经网络的训练需要定义损失函数,来表示真实值和预测值之间的偏差,当偏差大于设定的阈值,将继续训练,反向传播更新网络参数来优化网络模型,直到损失函数达到极小值或者小于设定的阈值。
[0106]
分割网络常使用dice loss来作为损失函数,表达式如下式所示:
[0107]
dice loss=1

dice
[0108]
其中,dice系数是一种集合相似度度量函数,用于计算两个样本的相似度,表达式如下:
[0109][0110]
x和y分别代表真实值和预测值,在分割任务中,样本中的元素即每一个像素点,dice系数衡量了各类别真实掩膜图(即步骤a2中标注之后的真假腔图像的掩膜图)和预测掩膜图(主动脉夹层真假腔图像分割网络的输出值,即预测值)之间的相似度,也是预测准确性的衡量指标。
[0111]
在实际的主动脉夹层临床图片中,常会有真假腔类别不平衡的情况存在,此时体积占比小的类别其分割效果不如体积占比大的类别,本发明针对的主动脉夹层类别包含a型和b型,其中a型夹层存在的类别不平衡问题更为明显。并且,在本发明中,小类别是真腔或者假腔是完全随机的,并不能通过施加固定的权重来解决类别不平衡的问题。因此,本实施例针对主动脉夹层图像中真假腔类别不平衡的问题,提出了类间三维共享面损失函数,表达式如下式所示:
[0112][0113]
其中n
c
表示类别数,在此主动脉夹层真假腔的分割任务中,类别包括真腔、假腔和背景共3类,即n
c
=3。m
isb
为类间三维共享面度量矩阵,大小为n
c
×
n
c
,代表预测掩膜图的类间三维共享面度量矩阵,代表真实掩膜图的类间三维共享面度量矩阵,m
isb
(i,j)的计算方法如下:
[0114][0115]
其中s
i
和s
ij
分别代表类别i外平面大小和类别j与类别i之间共享面大小,在三维体数据中,每一个像素当作一个立方体,每一面面积为单位1,则对于空间中位于(x,y,z)位置的像素点,计算与其他类别的共享面时,要考虑周围相邻六个像素点,即临近像素点集m={(x,y

1,z),(x,y 1,z),(x

1,y,z),(x 1,y,z),(x,y,z

1),(x,y,z 1)}。
[0116]
在本实施例中n
c
=3,则有:
[0117][0118]
类间三维共享面度量矩阵不依赖于目标对象的大小,而对空间相邻共享面变化敏感,一定程度上平衡了小类别对dice loss贡献小的问题,提高了小类别的分割精度,解决了主动脉真假腔类别不平衡的问题。
[0119]
最终总损失为dice loss和isb loss的组合,如下式所示:
[0120][0121]
总体来讲,训练分割网络模型的过程包括:设定主动脉夹层真假腔分割神经网络的超参数,包括学习速率,迭代次数,批处理数量等,之后开始训练网络,将预处理后的真假腔数据输入主动脉真假腔图像分割网络,输出得到具有类别信息的真假腔分割掩膜图,再通过损失函数overallloss,计算真假腔分割掩膜图与标注之后的真假腔图像的掩膜图之间的损失函数,直到结果收敛,损失函数小于设定阈值,或者达到训练次数,否则反向更新网络的超参数,继续训练,反向更新一次网络的超参数即为一次训练。训练完成后,将网络模型保存下来,得到基于注意力机制的分割网络模型,以便后续使用。
[0122]
经过上述对主动脉真假腔图像分割网络训练后,该神经网络对于训练集和验证集具有一定的拟合能力,为考察模型对于新数据的拟合能力,本实施例使用测试集对上述基于注意力机制的分割网络模型进行性能评估,利用dice系数判断所述分割网络模型的性能是否达到要求;
[0123]
若各类别分割的dice系数均大于90%,则所述分割网络模型满足性能要求;
[0124]
若所述分割网络模型达不到性能要求,则调整所述超参数,重新训练网络模型。
[0125]
在具体使用该分割网络模型的过程中,首先使用步骤a2中数据预处理对原始的主动脉夹层cta图像进行处理,再加载保存下来的网络模型超参数,然后将预处理后的数据输
入到该分割网络模型中,通过前向传播计算得到主动脉夹层真假腔图像分割结果掩膜图。
[0126]
最后将该分割结果掩膜图映射到原主动脉夹层cta图像中,即可得到对应的主动脉夹层真假腔在原图上的预测分割结果。
[0127]
如图5所示,本发明实施例还提供了一种全类型主动脉夹层真假腔图像分割系统,所述系统包括:
[0128]
图像获取单元m1,用于获取待分割真假腔图像;
[0129]
图像分割单元m2,用于利用基于注意力机制的分割网络模型对所述待分割真假腔图像进行分割,得到分割结果掩膜图;
[0130]
分割结果获取单元m3,用于将所述分割结果掩膜图映射到所述待分割真假腔图像中,得到分割结果。
[0131]
作为一种可选的实施方式,本实施例所提供的全类型主动脉夹层真假腔图像分割系统还包括:
[0132]
训练图像获取单元,用于获取真假腔图像;
[0133]
训练图像标注单元,用于对所述真假腔图像中的主动脉区域及真腔和假腔进行标注,得到真假腔数据;所述真假腔数据包括真假腔图像和标注之后的真假腔图像的掩膜图;
[0134]
分割网络模型训练单元,用于:
[0135]
设定分割神经网络的超参数,利用3d卷积提取真假腔数据的主动脉夹层空间相关性,并利用混合注意力模块作为基本单元进行上下采样对所述真假腔数据施加权重,通过指数函数归一化完成每一像素值类别的预测,得到具有类别信息的真假腔分割掩膜图;计算所述真假腔分割掩膜图与所述标注之后的真假腔图像的掩膜图之间的损失函数,直至结果收敛或者达到训练次数,得到分割网络模型。
[0136]
本发明实施例提供的全类型主动脉夹层真假腔图像分割方法及系统针对部分主动脉夹层假腔ct值与其他组织较接近,在边界处难以区分的问题,提出使用对数灰度变化的处理过程,可以强化主动脉夹层数据特征,提高模型精度。通过使用对数灰度变换对图像进行处理,根据对数函数的性质,在靠近0的小值取对数后差距会增大,并且对大值的增长也有一定的限制。故使用对数变换后可以增大低ct值区域的差别,使之原本难以区分的边界更为明显,使模型更容易学习到主动脉夹层的特征,从而模型精度提升。
[0137]
本发明针对主动脉夹层真假腔具有空间相关性的特点,提出使用基于注意力机制的主动脉夹层真假腔分割神经网络模型,使模型对于真假腔的区别更加敏感,从而提高了模型的精度。通过使用混合注意力模块,在训练中原三维体数据被施加权重并逐步优化此权重,使重要部分的权重上升,模型更能关注到决定主动脉真假腔的关键部位,加强了模型对于空间相关性的敏感度,增加模型分割准确度。
[0138]
本发明针对全类型主动脉真假腔分割的类别不平衡问题,提出应用类间三维共享面损失提高小类别对于训练过程的贡献,使模型对于真假腔类别不平衡的样本也能具有很好的拟合能力。类间三维共享面度量函数不依赖于目标对象的大小,而对空间相邻共享面变化敏感,一定程度上平衡了体积小的类别对基于像素点评估的dice loss贡献小的问题,提高了小类别的分割精度,解决了主动脉真假腔类别不平衡分割精度低的问题,使模型同时适用于a型和b型主动脉夹层的真假腔分割。
[0139]
对于实施例公开的系统而言,由于其与实施例公开的方法相对应,所以描述的比
较简单,相关之处参见方法部分说明即可。
[0140]
本文中应用了具体个例对本发明的原理及实施方式进行了阐述,以上实施例的说明只是用于帮助理解本发明的方法及其核心思想;同时,对于本领域的一般技术人员,依据本发明的思想,在具体实施方式及应用范围上均会有改变之处。综上所述,本说明书内容不应理解为对本发明的限制。
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献