用于多分段血管实时自动分割方法、装置、设备及介质

未命名 08-25 阅读:145 评论:0


1.本发明涉及血管分割技术领域,尤其涉及一种用于多分段血管实时自动分割方法、装置、设备及介质。


背景技术:

2.血管介入技术能够有效治疗血管疾病,具有创伤小、反应低、恢复快的优点,还具有靶向性特点,可使部分不能耐受手术或失去手术机会或耐药患者得到有效治疗,在部分领域已经取代外科手术成为首选治疗方式。
3.然而,这一任务目前面临着以下难点:(1)由于dsa(digital subtraction angiography,数字减影血管造影)的低质量成像以及影像中腹部内容物的伪影造成血管边界相对模糊;(2)造影剂随血流高速扩散,导致其分布不均匀,在部分血管中像素灰度差异较大,无法判断血管的准确边界;(3)不同于自然图像,血管分段之间没有明确的分界线,因此将不同分段的血管准确分割也是挑战性问题。


技术实现要素:

4.本发明提供一种用于多分段血管实时自动分割方法、装置、设备及介质,用以解决现有技术中dsa图像中血管分段之间没有明确的分界线,导致血管分割的准确率低的缺陷。
5.本发明提供一种用于多分段血管实时自动分割方法,包括:
6.获取待分割图像;
7.将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;
8.所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;
9.所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;
10.所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;
11.所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
12.根据本发明提供的一种用于多分段血管实时自动分割方法,所述类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络;
13.所述图像特征提取网络用于对所述图像特征进行语义特征提取,得到深层特征,所述类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取特征,所述融合网络用于对所述深层特征和所述类别先验提取特征进行融合,得到类别先验特征。
14.根据本发明提供的一种用于多分段血管实时自动分割方法,所述主干网络包括编
码部分、解码部分和若干全局先验注意力模块;
15.所述编码部分与所述解码部分连接,所述编码部分包括多个串联的编码器,所述解码部分包括多个串联的解码器;
16.所述全局先验注意力模块用于对所述编码器输出的图像特征进行全局先验注意力提取,得到全局先验注意力特征;
17.所述解码器用于对前一解码器输出的前一解码结果和对应的全局先验注意力模块输出的全局先验注意力特征进行解码,得到当前解码结果。
18.根据本发明提供的一种用于多分段血管实时自动分割方法,所述全局先验注意力模块包括第一分支和第二分支;
19.所述第一分支用于提取所述图像特征的隐藏层特征和各类别血管的语义特征,并基于所述隐藏层特征和所述各类别血管的语义特征提取第一注意力特征,并基于所述第一注意力特征和所述各类别血管的语义特征,得到第二注意力特征;
20.所述第二分支用于基于所述第二注意力特征和所述图像特征确定所述全局先验注意力特征。
21.根据本发明提供的一种用于多分段血管实时自动分割方法,所述结构先验网络包括稀疏掩码分支和卷积分支;
22.所述稀疏掩码分支用于对所述特征图进行卷积操作和聚合操作,得到稀疏掩码特征,所述卷积分支用于对所述特征图进行卷积操作,得到卷积特征,并基于所述卷积特征和所述稀疏掩码特征确定所述结构先验特征。
23.根据本发明提供的一种用于多分段血管实时自动分割方法,所述血管分割模型的训练步骤包括:
24.获取初始血管分割模型、样本图像以及所述样本图像的标签真值图像;
25.将所述样本图像输入至初始血管分割模型,得到所述初始血管分割模型输出的血管预测分割结果、所述样本图像的图像特征、所述样本图像的稀疏掩码特征,并基于所述初始血管分割模型的主干网络中的融合卷积模块,对所述样本图像的图像特征进行融合卷积,得到融合卷积特征;
26.基于所述血管预测分割结果与所述样本图像的标签真值图像之间的差异、所述样本图像的稀疏掩码特征与所述样本图像的标签真值图像之间的差异,以及所述样本图像的融合卷积特征与所述样本图像的标签真值图像之间的差异,对所述初始血管分割模型进行参数迭代,得到所述血管分割模型。
27.本发明还提供一种用于多分段血管实时自动分割装置,包括:
28.获取单元,用于获取待分割图像;
29.血管分割单元,用于将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;
30.所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;
31.所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;
32.所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得
到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;
33.所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
34.本发明还提供一种电子设备,包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现如上述任一种所述用于多分段血管实时自动分割方法。
35.本发明还提供一种非暂态计算机可读存储介质,其上存储有计算机程序,该计算机程序被处理器执行时实现如上述任一种所述用于多分段血管实时自动分割方法。
36.本发明还提供一种计算机程序产品,包括计算机程序,所述计算机程序被处理器执行时实现如上述任一种所述用于多分段血管实时自动分割方法。
37.本发明提供的用于多分段血管实时自动分割方法、装置、设备及介质,将待分割图像输入至血管分割模型,得到血管分割模型输出的血管分割结果,血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络,主干网络用于提取待分割图像的图像特征,并对图像特征进行解码得到特征图,类别先验网络用于基于类别先验知识,对图像特征进行血管分类,得到类别先验特征,结构先验网络用于基于结构先验知识,对特征图进行血管分类,得到结构先验特征,动态控制网络用于对类别先验特征、结构先验特征和图像特征进行特征融合,得到融合特征;主干网络还用于基于融合特征和特征图进行血管分割。类别先验网络和结构先验网络的结合可以有效地引导血管分割模型学习更好的数据表示,提高了血管分割的准确性和可靠性,并且,对于不同病人的复杂血管图像也具有很强的鲁棒性,且不需要预处理、人工操作就能得到准确的分割结果,自动化程度高,进一步提高了血管分割的实时响应性。
附图说明
38.为了更清楚地说明本发明或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作一简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
39.图1是本发明提供的用于多分段血管实时自动分割方法的流程示意图;
40.图2是本发明提供的血管分割模型的结构示意图;
41.图3是本发明提供的类别先验网络的结构示意图;
42.图4是本发明提供的全局先验注意力模块的结构示意图;
43.图5是本发明提供的推理时间和参数数量的示意图;
44.图6是本发明提供的用于多分段血管实时自动分割装置的结构示意图;
45.图7是本发明提供的电子设备的结构示意图。
具体实施方式
46.为使本发明的目的、技术方案和优点更加清楚,下面将结合本发明中的附图,对本发明中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳
动前提下所获得的所有其他实施例,都属于本发明保护的范围。
47.本发明的说明书和权利要求书中的术语“第一”、“第二”等是用于区别类似的对象,而不用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便本技术的实施例能够以除了在这里图示或描述的那些以外的顺序实施,且“第一”、“第二”等所区分的对象通常为一类。
48.相关技术中,腹部及胸部血管包括髂动脉、腹主动脉、双肾动脉、肠系膜上动脉以及胸主动脉。动脉瘤以及狭窄闭塞是这些血管的主要病变,例如腹主动脉瘤、胸主动脉夹层、肾动脉闭塞等。随着人口老龄化以及高血压、动脉硬化等高危因素的增加,主动脉瘤以及主动脉夹层的发病率也随之增高,严重威胁着人类的生命健康,给诸多患者带来无限的痛苦和生命危险。这些主动脉疾病大多病情凶险,致死率和致残率较高,治疗难度较大。
49.腹主动脉瘤是腹主动脉在动脉粥样硬化、创伤、感染等各种病理因素的作用下,以局部扩张向外膨出为主要表现的大血管疾病。腹主动脉瘤的患病率占主动脉瘤的63%-79%。瘤体可能产生血栓,引起远端肢体缺血坏死,严重可能导致肢体截肢。瘤体一旦破裂,死亡率高达50%-80%。
50.介入手术治疗是近年迅速发展起来的一种融合了影像判断和临床治疗于一体的治疗方法。它是在数字减影血管造影机、ct(computed tomography,电子计算机断层扫描)、超声和磁共振等影像设备的引导和监视下,利用穿刺针、导管及其他介入器材,通过人体自然孔道或微小的创口将支架等特定的手术器械导入人体病变部位进行微创治疗的一系列技术的总称。目前已经成为与传统的内科、外科手术并列的临床三大支柱性治疗学科。血管介入技术能够有效治疗血管疾病,具有创伤小、反应低、恢复快的优点,还具有靶向性特点,可使部分不能耐受手术或失去手术机会或耐药患者得到有效治疗,在部分领域已经取代外科手术成为首选治疗方式。首先,常见的用于血管介入治疗的腹部以及胸部血管疾病包括主动脉瘤、主动脉夹层、髂动脉或股动脉狭窄与闭塞、肾动脉狭窄等。
51.血管介入手术的做法具体做法为:第一,在术前cta(computed tomographic angiography,计算机断层扫描血管造影)图像中明确血管的病变严重程度以及血管大致形状,制定相关手术计划。
52.第二,明确介入血管的入路,临床上常用的有股动脉入路以及桡动脉穿刺入路,并且明确入路直径。
53.第三,明确入路之后,要在导丝、导管的引导下超选病变血管,通过病变血管抄选后,完成相应的血管造影,以明确病变的部位、性质以及病变的程度,同时评估是否需要介入治疗或是选择开刀治疗。
54.最后,根据病变的程度以及病变的性质选择不同的支架来进行支架植入术。
55.基于上述步骤的血管介入手术无需开胸以及体外循环,引起的中枢系统并发症也比较少,因此微创血管介入手术在最近三十年中得到了迅速发展。但同时该手术也存在一些明显的不足:1)微创血管介入手术中需要使用x射线引导,正对病人胸部。这意味着医护人员和患者都在处于高辐射的x射线的照射环境下。每次介入手术大概要一个小时左右,而复杂的手术则需要四到五个小时。医护人员长时间在如此恶劣环境下进行手术,身体受到的辐射非常大。2)人体的血管直径一般在5mm以下,所以在微创血管介入手术过程中医生会感到非常疲劳。由此会引起医生手部颤抖、肌肉神经的反馈、眼部视力模糊,导致动作不准
确。这些因素都将直接影响微创血管介入手术的质量和手术精度,降低患者术后的生存质量以及并发症的增加。此外器械在血管中的任何失误和重复都有可能增加人体血管壁的损伤。3)微创血管介入手术的过程非常繁杂,医生需要经过大量的训练才能够掌握实际手术操作能力。在手术过程中,经常会出现x射线影像的模糊以及导丝在人体内的某些狭小的、分支数目较多的血管递送的情况,这些情况都需要医生有很好的观察力和高超的技术。
56.目前机器人和计算机辅助的一些技术已经与临床紧密结合,从而提高医生操作的敏捷度和准确度。作为机器人辅助介入手术的必要环节,实时且全自动的多分段血管分割的方法可以为医生们提供必要的视觉和触觉反馈辅助。
57.然而,这一任务目前面临着以下难点:(1)由于dsa影像的低质量成像以及影像中腹部内容物的伪影造成血管边界相对模糊;(2)造影剂随血流高速扩散,导致其分布不均匀,在部分血管中像素灰度差异较大,无法判断血管的准确边界;(3)不同于自然图像,血管分段之间没有明确的分界线,因此将不同分段的血管准确分割也是挑战性问题。
58.并且,血管的形状极其不规则,同一张造影图像中的血管的粗细差异很大,并且通常具有多个分支。同时,每个人的血管成像有极大的差异性。这些都对血管的准确分割造成了困难。关于血管分割算法的研究有很多。根据成像的维度可以分为三维血管分割(cta或者mra(magnetic resonance angiography,磁共振血管成像)和二维血管分割(dsa/超声)。早期血管分割方法主要是基于跟踪轨迹的方法以及基于模型变换的方法,这些方法需要人工设计的特征,并且无法保证血管分割的有效性。
59.基于上述问题,本发明提供一种用于多分段血管实时自动分割方法,图1是本发明提供的用于多分段血管实时自动分割方法的流程示意图,如图1所示,该方法包括:
60.步骤110,获取待分割图像。
61.具体地,可以获取待分割图像,此处的待分割图像是指后续用于血管分割的图像,此处的待分割图像可以是dsa(digital subtraction angiography,数字减影血管造影)图像等,例如,待分割图像可以是腹部血管的dsa影像,也可以是脑血管的dsa影像,还可以是冠状动脉的dsa影像等,本发明实施例对此不作具体限定。
62.例如,当待分割图像是腹部血管的dsa影像时,腹部血管可以包括六种关键血管分段:双肾动脉(renal artery,ra)、肠系膜上动脉(superior mesenteric artery,sma)、肾动脉至腹主动脉瘤分段(renal artery to abdominal aortic aneurysm,ra-aaa)、腹主动脉瘤(abdominal aortic aneurysm,aaa)、腹主动脉瘤至髂动脉分段(abdominal aortic aneurysm to common iliac artery,aaa-cia)和髂总动脉(common iliac artery,cia)。
63.步骤120,将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;
64.所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;
65.所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;
66.所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;
67.所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
68.具体地,图2是本发明提供的血管分割模型的结构示意图,如图2所示,在获取到待分割图像之后,可以将待分割图像输入至血管分割模型,得到血管分割模型输出的血管分割结果。
69.此处的血管分割模型可以包括主干网络、类别先验网络、结构先验网络和动态控制网络,此处的主干网络用于提取待分割图像的图像特征,并对图像特征进行解码得到特征图。主干网络可以包括编码部分、解码部分、若干全局先验注意力模块以及融合卷积模块。
70.此处的全局先验注意力模块用于提高主干网络对于特征的细化,使得主干网络能够通过更加关注dsa影像所特有的先验知识来强化特征表达,即获取不同类别的血管分段之间具有的空间连接关系。
71.其中,编码部分可以包括6个卷积模块,每个卷积模块可以包括两个连接的卷积操作,包括一个3
×
3卷积、batch norm操作和relu(rectified linear unit)函数。解码部分可以包括5个解码模块,也可以包括4个解码模块等,本发明实施例对此不作具体限定。
72.例如,可以将一个大小为512
×
512
×
3的待分割图像输入至主干网络进行六次卷积操作实现下采样和特征提取,经过编码器后的特征图的分辨率降为64
×
64
×
512。编码器之后连接一个融合卷积模块进一步提取图像特征,同时利用融合卷积操作之后的特征图输入到对称的解码器中进行上采样和特征融合操作,将特征图的分辨率恢复到512
×
512
×
3,由此,提高特征映射的分辨率,压缩通道数。
73.此处的类别先验网络用于基于类别先验知识,对图像特征进行血管分类,得到类别先验特征,此处的类别先验知识是指待分割图像中特有的不同类别分段之间的概率关系,此处的类别先验特征反映了图像特征中类别分割层面的特征信息。
74.此处的类别先验网络可以包括图像特征提取网络、类别先验提取网络和融合网络。此处的图像特征提取网络可以包括聚合层和池化层,此处的类别先验提取网络可以包括卷积层,此处的卷积层可以是gcn(graph convolutional neural networks,图卷积神经网络),也可以是级联结构的多层卷积神经网络(convolutional neural network,cnn),还可以是深度神经网络(deep neural networks,dnn)等,本发明实施例对此不作具体限定。
75.此处的结构先验网络用于基于结构先验知识,对特征图进行血管分类,得到结构先验特征,此处的结构先验知识是指结构层面的先验知识,此处的结构先验特征反映了结构层面的特征信息。此处的结构先验网络可以包括稀疏掩码分支和卷积分支。此处的稀疏掩码分支可以包括argmax层和卷积层。
76.为了更好地利用类别先验特征以及结构先验特征,动态控制网络用于对类别先验特征、结构先验特征和图像特征进行特征融合,得到融合特征。此处,对类别先验特征、结构先验特征和图像特征进行特征融合,可以是将类别先验特征、结构先验特征和图像特征进行拼接,还可以是利用注意力机制将类别先验特征、结构先验特征和图像特征进行加权之后再拼接,本发明实施例对此不作具体限定。
77.此外,在对类别先验特征、结构先验特征和图像特征进行特征融合之前,还可以基于池化层对图像特征进一步提取特征信息,此处的池化层可以是全局最大池化层(global max pooling,gmp),也可以是平均池化层(average pooling)等,本发明实施例对此不作具
体限定。
78.动态控制网络获得最终动态分割层的卷积参数,该动态分割层包含3个连续的卷积层,每个卷积层有1
×
1核和偏差。动态控制网络对类别先验特征、结构先验特征和图像特征进行特征融合,然后针对动态分割头动态生成3层卷积核参数w1,w2,w3。
79.此处的主干网络还用于基于融合特征和特征图进行血管分割,此处,基于融合特征和特征图进行血管分割,可以先对融合特征和特征图进行特征融合,再基于特征融合后的特征进行血管分割。
80.此处,对融合特征和特征图进行特征融合,可以是将融合特征和特征图进行拼接,还可以是利用注意力机制将融合特征和特征图进行加权之后再拼接,本发明实施例对此不作具体限定。
81.此处,基于融合特征和特征图进行血管分割的公式如下:
[0082][0083]
其中,分别为结构先验特征、类别先验特征和图像特征的卷积核参数,m为特征图。
[0084]
本发明实施例提供的方法,将待分割图像输入至血管分割模型,得到血管分割模型输出的血管分割结果,血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络,主干网络用于提取待分割图像的图像特征,并对图像特征进行解码得到特征图,类别先验网络用于基于类别先验知识,对图像特征进行血管分类,得到类别先验特征,结构先验网络用于基于结构先验知识,对特征图进行血管分类,得到结构先验特征,动态控制网络用于对类别先验特征、结构先验特征和图像特征进行特征融合,得到融合特征;主干网络还用于基于融合特征和特征图进行血管分割。类别先验网络和结构先验网络的结合可以有效地引导血管分割模型学习更好的数据表示,提高了血管分割的准确性和可靠性,并且,对于不同病人的复杂血管图像也具有很强的鲁棒性,且不需要预处理、人工操作就能得到准确的分割结果,自动化程度高,进一步提高了血管分割的实时响应性。
[0085]
基于上述实施例,图3是本发明提供的类别先验网络的结构示意图,如图3所示,所述类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络;
[0086]
所述图像特征提取网络用于对所述图像特征进行语义特征提取,得到深层特征,所述类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取特征,所述融合网络用于对所述深层特征和所述类别先验提取特征进行融合,得到类别先验特征。
[0087]
具体地,类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络,此处的图像特征提取网络可以包括多个聚合层和池化层,此处的类别先验提取网络可以包括多个卷积层,此处的卷积层可以是gcn(graph convolutional neural networks,图卷积神经网络),也可以是级联结构的多层卷积神经网络(convolutional neural network,cnn),还可以是深度神经网络(deep neural networks,dnn)等,本发明实施例对此不作具体限定。
[0088]
图像特征提取网络用于对图像特征进行语义特征提取,得到深层特征,类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取
特征。
[0089]
此处的类别先验知识是指待分割图像中特有的不同类别分段之间的概率关系,此处的类别先验提取特征反映了图像特征中类别分割层面提取到的特征信息。
[0090]
图像特征提取网络的聚合层提取主干网络的编码部分中的四个卷积模块中的特征图进行特征融合,获得图像中的高级语义信息,其具体的实现方式可以用公式表示为:
[0091][0092]
其中,使用表示从编码器中提取的特征映射,x4表示深层特征,表示对应的1
×
1卷积,表示下采样卷积,concat是通道级连接。是通过全局最大池化(gmp)从z4导出的聚合特征。
[0093]
同时,利用dsa血管影像的特征,即造影剂由上至下流动,使得血管分段在影像中的出现具有一定的概率关系,来构建图卷积网络中的邻接矩阵a。此外,使用词向量代表不同的血管类别,使用glove(global vectors for word representation)方法训练得到初始血管分类特征表示多分支血管的分布特征,其中n是类别数,m是类级词嵌入的维数,m可以设置为300。从而,类别先验提取特征e
i+1
可以表示为:
[0094]ei+1
=g(aeiwi)
[0095]
其中,i∈(1,2)表示gcn层数,e2是最终生成的相关图。最终,融合网络对获取的深层特征以及经过图卷积提取的类别先验提取特征进行特征融合,得到类别先验特征vc:
[0096]vc
=z
*
×
(e2)
t
[0097]
本发明实施例提供的方法,类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络,图像特征提取网络用于对图像特征进行语义特征提取,得到深层特征,类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取特征,融合网络用于对深层特征和类别先验提取特征进行融合,得到类别先验特征,提高了类别先验特征提取的准确性和可靠性,进一步提高了后续血管分割的准确性和可靠性。
[0098]
基于上述实施例,所述主干网络包括编码部分、解码部分和若干全局先验注意力模块;
[0099]
所述编码部分与所述解码部分连接,所述编码部分包括多个串联的编码器,所述解码部分包括多个串联的解码器;
[0100]
所述全局先验注意力模块用于对所述编码器输出的图像特征进行全局先验注意力提取,得到全局先验注意力特征;
[0101]
所述解码器用于对前一解码器输出的前一解码结果和对应的全局先验注意力模块输出的全局先验注意力特征进行解码,得到当前解码结果。
[0102]
具体地,主干网络可以包括编码部分、解码部分和若干全局先验注意力模块,此处的编码部分与解码部分连接,编码部分包括多个串联的编码器,解码部分包括多个串联的解码器。例如,如图2所示,编码部分包括6个卷积模块,解码部分包括5个解码模块。
[0103]
全局先验注意力模块用于对编码器输出的图像特征进行全局先验注意力提取,得到全局先验注意力特征。即,此处的全局先验注意力模块用于提高主干网络对于特征的细化,使得主干网络能够通过更加关注dsa影像所特有的先验知识来强化特征表达,即获取不
同类别的血管分段之间具有的空间连接关系。
[0104]
如图2所示,解码器用于对前一解码器输出的前一解码结果和对应的全局先验注意力模块输出的全局先验注意力特征进行解码,得到当前解码结果。
[0105]
基于上述实施例,图4是本发明提供的全局先验注意力模块的结构示意图,如图4所示,所述全局先验注意力模块包括第一分支和第二分支;
[0106]
所述第一分支用于提取所述图像特征的隐藏层特征和各类别血管的语义特征,并基于所述隐藏层特征和所述各类别血管的语义特征提取第一注意力特征,并基于所述第一注意力特征和所述各类别血管的语义特征,得到第二注意力特征;
[0107]
所述第二分支用于基于所述第二注意力特征和所述图像特征确定所述全局先验注意力特征。
[0108]
具体地,考虑到全局先验注意力模块主要用来融合dsa血管影像所拥有的特征,即不同类别的血管分段之间具有空间连接关系,这种空间连接关系在普通的cnn特征提取过程中无法获得,因此,利用这种空间连接关系构建了用于图卷积的邻接矩阵ac。ac是一个对角元素均为1的对称矩阵,如果第i个和第j个对象是空间连接的,则否则,ac表示七个类的空间连接(包括血管和背景的6个子部分,c1为ra,c2为sma,c3为ra-aaa,c4为aaa,c5为aaa-cia,c6为cia,bg为背景(background),即c=7)。
[0109]
全局先验注意力模块可以包括第一分支和第二分支,此处的第一分支用于提取图像特征的隐藏层特征和各类别血管的语义特征,并基于隐藏层特征和各类别血管的语义特征提取第一注意力特征。
[0110]
编码器的残差块输出的图像特征用表示,其中,n和h
×
w分别表示图像特征的分辨率和通道数。f首先同时经过两个带有n和c通道数的1
×
1卷积层,以生成fh和fc两个特征映射。表示图像特征的隐藏层特征,聚合了f中包含的每个类的语义信息,即表示各类别血管的语义特征。然后,fc和fh分别被重塑为和随后,在初始嵌入的初始原型可计算如下:
[0111][0112]
在初始嵌入后,根据上面的相邻矩阵ac采用gcn层(graph convolutional neural networks,图卷积神经网络)来模拟不同解剖原型之间的空间关联。每个gcn层采用卷积运算,则第一注意力特征为:
[0113][0114]
其中,是要学习的第i层的变换矩阵,表示非线性函数,i=[1,

,l]。
[0115]
此处的第一分支可以基于第一注意力特征和各类别血管的语义特征,得到第二注意力特征,此处的第二注意力特征的公式如下:
[0116][0117]
此处的第二分支用于基于第二注意力特征和图像特征确定全局先验注意力特征,此处的全局先验注意力特征反映了全局先验层面的特征信息,则全局先验注意力特征的公式为:
[0118][0119]
其中,γ是一个可学习的参数,作为剩余操作的大小因子。
[0120]
然后,可以将经过细化的特征映射(全局先验注意力特征)发送到解码器进行解码。
[0121]
本发明实施例提供的方法,全局先验注意力模块主要用来融合dsa血管影像所拥有的特征,即不同类别的血管分段之间具有空间连接关系,从而获取了在普通的cnn特征提取过程中无法获得的空间连接关系,提高了全局先验注意力特征的准确性和可靠性。
[0122]
基于上述实施例,所述结构先验网络包括稀疏掩码分支和卷积分支;
[0123]
所述稀疏掩码分支用于对所述特征图进行卷积操作和聚合操作,得到稀疏掩码特征,所述卷积分支用于对所述特征图进行卷积操作,得到卷积特征,并基于所述卷积特征和所述稀疏掩码特征确定所述结构先验特征。
[0124]
具体地,结构先验网络可以包括稀疏掩码分支和卷积分支,此处的稀疏掩码分支用于对特征图m进行卷积操作和聚合操作,得到稀疏掩码特征,此处的稀疏掩码特征反映了结构层面的特征信息。卷积分支用于对特征图进行卷积操作,得到卷积特征。
[0125]
即,结合解码器最后一层所得到的富含高级语义信息和细节特征的特征图,可以获取结构先验特征。具体来说,在特征图通过1
×
1卷积模块和聚合操作(聚合操作可以为argmax操作)后,可以得到一个稀疏掩码特征同时,从特征图m中进一步提取了卷积特征其中,n表示类的数量。
[0126]
然后,mc和mf分别被重塑为和
[0127]
最后,基于卷积特征和稀疏掩码特征确定结构先验特征vs为:
[0128][0129]
基于上述实施例,所述血管分割模型的训练步骤包括:
[0130]
步骤210,获取初始血管分割模型、样本图像以及所述样本图像的标签真值图像;
[0131]
步骤220,将所述样本图像输入至初始血管分割模型,得到所述初始血管分割模型输出的血管预测分割结果、所述样本图像的图像特征、所述样本图像的稀疏掩码特征,并基于所述初始血管分割模型的主干网络中的融合卷积模块,对所述样本图像的图像特征进行融合卷积,得到融合卷积特征;
[0132]
步骤230,基于所述血管预测分割结果与所述样本图像的标签真值图像之间的差异、所述样本图像的稀疏掩码特征与所述样本图像的标签真值图像之间的差异,以及所述样本图像的融合卷积特征与所述样本图像的标签真值图像之间的差异,对所述初始血管分割模型进行参数迭代,得到所述血管分割模型。
[0133]
具体地,为了更好地提高血管分割模型的分割性能,可以通过如下步骤获取血管分割模型:
[0134]
可以预先获取初始血管分割模型、样本图像以及样本图像的标签真值图像,此处的初始血管分割模型即训练血管分割模型的初始模型,初始血管分割模型的参数可以是预先设置的,也可以是随机生成的,本发明实施例对此不作具体限定。
[0135]
在得到初始血管分割模型后,即可应用预先收集好的样本图像以及样本图像的标签真值图像,对初始血管分割模型进行训练:
[0136]
首先,将样本图像输入至初始血管分割模型,得到初始血管分割模型输出的血管预测分割结果,并由初始血管分割模型中的主干网络的编码部分输出样本图像的图像特征,由初始血管分割模型中的结构先验网络中的稀疏掩码分支输出样本图像的稀疏掩码特征。
[0137]
然后,可以基于初始血管分割模型的主干网络中的融合卷积模块,对样本图像的图像特征进行融合卷积,得到融合卷积特征。
[0138]
在得到血管预测分割结果、稀疏掩码特征和融合卷积特征之后,即可将血管预测分割结果与样本图像的标签真值图像进行比较,根据此两者之间的差异计算得到第一损失函数值,并可以将样本图像的稀疏掩码特征与样本图像的标签真值图像进行比较,根据此两者之间的差异计算得到第二损失函数值,以及可以将样本图像的融合卷积特征与样本图像的标签真值图像进行比较,根据此两者之间的差异计算得到第三损失函数值。
[0139]
可以理解的是,血管预测分割结果与样本图像的标签真值图像之间的差异越大,第一损失函数值越大;血管预测分割结果与样本图像的标签真值图像之间的差异越小,第一损失函数值越小。
[0140]
可以理解的是,样本图像的稀疏掩码特征与所述样本图像的标签真值图像之间的差异越大,第二损失函数值越大;样本图像的稀疏掩码特征与所述样本图像的标签真值图像之间的差异越小,第二损失函数值越小。
[0141]
可以理解的是,样本图像的融合卷积特征与所述样本图像的标签真值图像之间的差异越大,第三损失函数值越大;样本图像的融合卷积特征与所述样本图像的标签真值图像之间的差异越小,第三损失函数值越小。
[0142]
在得到第一损失函数值、第二损失函数值和第三损失函数值之后,可以基于第一损失函数、第二损失函数值和第三损失函数值对初始血管分割模型进行参数迭代,也可以基于第一损失函数、第二损失函数值和第三损失函数值的加权之和对初始血管分割模型进行参数迭代,并将参数迭代之后的初始血管分割模型作为血管分割模型。
[0143]
其中,基于第一损失函数、第二损失函数值和第三损失函数值的加权之和对初始血管分割模型进行参数迭代的公式如下:
[0144]
l
seg
=α1l
cel
(r1,g)+α2l
cel
(r2,g)+α3l
cel
(r3,g)
[0145]
其中,r1,r2,r3分别表示样本图像的稀疏掩码特征、样本图像的融合卷积特征和血管预测分割结果,g表示样本图像的标签真值图像,α1,α2,α3表示超参数,α1,α2,α3可以设置为0.2,0.3,1.0.
[0146]
其中,l
cel
表示融合交叉熵损失函数和dice损失函数的损失函数,l
cel
的公式如下:
[0147]
l
cel
=(1-α)h-αlog(d)
[0148][0149]
[0150]
其中,d是dice系数损失函数,h是交叉熵损失函数,调制因子α在α≥0的范围内可变。α是用于调整增强交叉熵损失函数和dice系数损失函数之间平衡的超参数,在本发明实施例中可以设为0.2。
[0151]
此处,可以使用dice系数损失函数(dice loss),也可以使用交叉熵损失函数(cross entropy loss function),还可以使用均方误差损失函数(mean squared error,mse)等对初始血管分割模型的参数进行更新,本发明实施例对此不作具体限定。
[0152]
即,在初始血管分割模型的训练过程中,学习到了对血管进行准确分割的能力。
[0153]
本发明实施例提供的方法,基于血管预测分割结果与样本图像的标签真值图像之间的差异、样本图像的稀疏掩码特征与样本图像的标签真值图像之间的差异,以及样本图像的融合卷积特征与样本图像的标签真值图像之间的差异,对初始血管分割模型进行参数迭代,得到血管分割模型,利用了多个中间输出特征的深监督训练方法,并且结合了dice系数损失函数和交叉熵损失函数的混合损失函数,可以有效解决类间数据不平衡的问题和错分类的问题,进一步提高了血管分割的准确性和可靠性。
[0154]
基于上述任一实施例,一种用于多分段血管实时自动分割方法,步骤如下:
[0155]
第一步,获取待分割图像。
[0156]
第二步,将待分割图像输入至血管分割模型,得到血管分割模型输出的血管分割结果。
[0157]
此处的血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络。
[0158]
主干网络用于提取待分割图像的图像特征,并对图像特征进行解码得到特征图。
[0159]
类别先验网络用于基于类别先验知识,对图像特征进行血管分类,得到类别先验特征。类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络。图像特征提取网络用于对图像特征进行语义特征提取,得到深层特征,类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取特征。融合网络用于对深层特征和类别先验提取特征进行融合,得到类别先验特征。
[0160]
结构先验网络用于基于结构先验知识,对特征图进行血管分类,得到结构先验特征,动态控制网络用于对类别先验特征、结构先验特征和图像特征进行特征融合,得到融合特征。
[0161]
主干网络还用于基于融合特征和特征图进行血管分割。主干网络包括编码部分、解码部分和若干全局先验注意力模块。
[0162]
编码部分与解码部分连接,编码部分包括多个串联的编码器,解码部分包括多个串联的解码器。全局先验注意力模块用于对编码器输出的图像特征进行全局先验注意力提取,得到全局先验注意力特征。
[0163]
解码器用于对前一解码器输出的前一解码结果和对应的全局先验注意力模块输出的全局先验注意力特征进行解码,得到当前解码结果。
[0164]
此处的全局先验注意力模块包括第一分支和第二分支,第一分支用于提取图像特征的隐藏层特征和各类别血管的语义特征,并基于隐藏层特征和各类别血管的语义特征提取第一注意力特征,并基于第一注意力特征和各类别血管的语义特征,得到第二注意力特征。
[0165]
此处的第二分支用于基于第二注意力特征和图像特征确定全局先验注意力特征。
[0166]
此处的结构先验网络包括稀疏掩码分支和卷积分支,此处的稀疏掩码分支用于对特征图进行卷积操作和聚合操作,得到稀疏掩码特征,卷积分支用于对特征图进行卷积操作,得到卷积特征,并基于卷积特征和稀疏掩码特征确定结构先验特征。
[0167]
此处的血管分割模型的训练步骤包括:
[0168]
首先,获取初始血管分割模型、样本图像以及样本图像的标签真值图像。
[0169]
然后,将样本图像输入至初始血管分割模型,得到初始血管分割模型输出的血管预测分割结果、样本图像的图像特征、样本图像的稀疏掩码特征,并基于初始血管分割模型的主干网络中的融合卷积模块,对样本图像的图像特征进行融合卷积,得到融合卷积特征。
[0170]
最后,基于血管预测分割结果与样本图像的标签真值图像之间的差异、样本图像的稀疏掩码特征与样本图像的标签真值图像之间的差异,以及样本图像的融合卷积特征与样本图像的标签真值图像之间的差异,对初始血管分割模型进行参数迭代,得到血管分割模型。
[0171]
此外,本发明实施例建立了多分支血管分割数据集,包括医院100例患者共1551张腹部dsa图像。每张图像的平均分辨率为1024
×
1024。使用labelme标注工具准确标注了肾动脉(ra)、肠系膜上动脉(sma)、肾动脉至腹主动脉瘤(ra-aaa)、腹主动脉瘤(aaa)、腹主动脉瘤至髂总动脉(aaa-cia)、髂总动脉(cia)六类血管。注释过程经过了临床专家的严格审查和指导。为了便于后续处理和分析,所有图像和标签都被调整为512
×
512像素。将数据集随机划分为训练集和测试集,分别包含1243张和308张图像。本发明实施例中将提出的用于多分段血管实时自动分割方法在建立的新数据集上进行了测试。
[0172]
在本研究中,使用miou(mean intersection over union)、95%hd95(hausdorff distance)和dice评分作为评价分割性能的指标。miou是一个基于区域的评估指标,通过测量每个区域中预测的分割与地面真相之间的重叠来计算分割的平均精度。hd95通过计算预测边界与地面真实边界之间的最大距离来评估分割边界的质量。最后,dice评分通过计算两者之间的重叠率来衡量预测分割和真值之间的相似性。
[0173]
本发明实施例中所提出的血管分割模型使用pytorch框架实现,并在ubuntu 20.04.1平台上使用两个nvidia-a6000显卡进行训练。本发明实施例采用adam优化器进行训练,初始学习率为7e-5
,权值衰减为0.5,动量为0.999。为了达到最佳性能,学习率每20个epoch降低0.9个因子。最大epoch步数设置为200,所有模型的批量大小固定为4。此外,在训练阶段使用了数据增强技术,包括随机旋转和随机垂直翻转。
[0174]
类别先验网络和结构先验网络在提高网络性能方面的效果通过表1得到了证明。在本研究中,将类型先验特征、结构先验特征和图像特征三个特征向量提取、合并作为动态控制器的输入,生成动态分割头的核。值得注意的是,结果表明,将类别先验网络和结构先验网络添加到主干网络中可以显著提高性能,当两个先验网络都被合并时,可以获得最好的结果。具体而言,结果表明,类别先验网络和结构先验网络的结合可以有效地引导网络学习更好的数据表示,最终提高性能。
[0175]
表1.实验结果
[0176][0177][0178]
在特征聚合过程中,对用于原始特征映射捕获的编码器块的数量进行了评估。具体来说,测试了从3到5的宽度,并分析了它们对类别先验网络的性能和鲁棒性的影响。结果如表2所示,类别先验网络的性能在宽度为4时趋于稳定。根据经验,由此确定特征聚合的宽度为4最适合。
[0179]
表2.不同数量的编码器的实验结果
[0180][0181]
在本发明实施例中,通过计算其平均dice得分和miou,将u-net中使用的一般跳过连接与所提出的全局先验注意力模块进行了比较。如表3所示,结果表明,与基线相比,所提出的全局先验注意力模块产生了1.48%的平均dice得分和0.93%的miou的增加,这表明所提出的全局先验注意力模块提供了比一般跳过连接的增量改进。
[0182]
表3.全局先验注意力模块的实验结果
[0183] mioumdicebaseline0.81900.8791with gpa0.82380.8939
[0184]
为了验证所提出的用于多分段血管实时自动分割方法在多分支血管分割任务中的有效性,主要实验是通过将pad-net与目前最先进的医学图像分割方法(包括u-net、u-net++、u-net3+、transunet和attention unet)进行比较。表4显示了在六个类别中分割血管的每个分段的性能指标,在hd95指标方面表现出更好的性能,表明更好的血管连接分割。此外,在难以分割的aaa-cia部分,用于多分段血管实时自动分割方法有了相当大的改进。
[0185]
表4.单类分割对比实验结果
[0186][0187][0188]
表4中血管分割模型用pad-net表示,表4中的结果表明,虽然大多数方法对容易分割的目标表现出良好的性能,但本发明实施例提出的方法对具有挑战性的目标取得了更好的结果。此外,pad-net在平均指标上表现优异,如表5所示,miou为82.83%,dice平均得分
为89.39%,hd95平均得分为23.21。这清楚地证明了本发明实施例所提供的方法对多分支血管分割任务的有效性。
[0189]
表5.多类分割对比实验结果
[0190]
methodsmioumdiceavg.hd95unet0.80700.823034.58unet++0.79110.845739.14unet3+0.77900.845793.92att.unet0.76320.837542.36transunet0.78080.848135.67pad-net0.82830.893923.21
[0191]
此外,为了对本发明实施例所提供的方法进行更全面和实时的评估,对推理时间和参数数量进行了计算。推理时间是基于300个空间大小切片的单个输入计算的。为了评估速度和准确性之间的权衡,比较了六种方法的性能,图5是本发明提供的推理时间和参数数量的示意图,如图5所示,血管分割模型用pad-net表示,与其他方法相比,该方法不仅具有轻量级和实时性,而且具有较高的分割精度和较少的参数。因此,血管分割模型具有更好的综合性能。
[0192]
下面对本发明提供的多分段血管实时自动分割装置进行描述,下文描述的多分段血管实时自动分割装置与上文描述的多分段血管实时自动分割方法可相互对应参照。
[0193]
基于上述任一实施例,本发明提供一种用于多分段血管实时自动分割装置,图6是本发明提供的用于多分段血管实时自动分割装置的结构示意图,如图6所示,该装置包括:
[0194]
获取单元610,用于获取待分割图像;
[0195]
血管分割单元620,用于将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;
[0196]
所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;
[0197]
所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;
[0198]
所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;
[0199]
所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
[0200]
本发明实施例提供的装置,将待分割图像输入至血管分割模型,得到血管分割模型输出的血管分割结果,血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络,主干网络用于提取待分割图像的图像特征,并对图像特征进行解码得到特征图,类别先验网络用于基于类别先验知识,对图像特征进行血管分类,得到类别先验特征,结构先验网络用于基于结构先验知识,对特征图进行血管分类,得到结构先验特征,动态控制网络用于对类别先验特征、结构先验特征和图像特征进行特征融合,得到融合特征;主干网络还用于基于融合特征和特征图进行血管分割。类别先验网络和结构先验网络的结合可以有效地引导血管分割模型学习更好的数据表示,提高了血管分割的准确性和可靠性,并且,对
于不同病人的复杂血管图像也具有很强的鲁棒性,且不需要预处理、人工操作就能得到准确的分割结果,自动化程度高,进一步提高了血管分割的实时响应性。
[0201]
基于上述任一实施例,所述类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络;
[0202]
所述图像特征提取网络用于对所述图像特征进行语义特征提取,得到深层特征,所述类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取特征,所述融合网络用于对所述深层特征和所述类别先验提取特征进行融合,得到类别先验特征。
[0203]
基于上述任一实施例,所述主干网络包括编码部分、解码部分和若干全局先验注意力模块;
[0204]
所述编码部分与所述解码部分连接,所述编码部分包括多个串联的编码器,所述解码部分包括多个串联的解码器;
[0205]
所述全局先验注意力模块用于对所述编码器输出的图像特征进行全局先验注意力提取,得到全局先验注意力特征;
[0206]
所述解码器用于对前一解码器输出的前一解码结果和对应的全局先验注意力模块输出的全局先验注意力特征进行解码,得到当前解码结果。
[0207]
基于上述任一实施例,所述全局先验注意力模块包括第一分支和第二分支;
[0208]
所述第一分支用于提取所述图像特征的隐藏层特征和各类别血管的语义特征,并基于所述隐藏层特征和所述各类别血管的语义特征提取第一注意力特征,并基于所述第一注意力特征和所述各类别血管的语义特征,得到第二注意力特征;
[0209]
所述第二分支用于基于所述第二注意力特征和所述图像特征确定所述全局先验注意力特征。
[0210]
基于上述任一实施例,所述结构先验网络包括稀疏掩码分支和卷积分支;
[0211]
所述稀疏掩码分支用于对所述特征图进行卷积操作和聚合操作,得到稀疏掩码特征,所述卷积分支用于对所述特征图进行卷积操作,得到卷积特征,并基于所述卷积特征和所述稀疏掩码特征确定所述结构先验特征。
[0212]
基于上述任一实施例,所述血管分割模型的训练步骤包括:
[0213]
获取初始血管分割模型、样本图像以及所述样本图像的标签真值图像;
[0214]
将所述样本图像输入至初始血管分割模型,得到所述初始血管分割模型输出的血管预测分割结果、所述样本图像的图像特征、所述样本图像的稀疏掩码特征,并基于所述初始血管分割模型的主干网络中的融合卷积模块,对所述样本图像的图像特征进行融合卷积,得到融合卷积特征;
[0215]
基于所述血管预测分割结果与所述样本图像的标签真值图像之间的差异、所述样本图像的稀疏掩码特征与所述样本图像的标签真值图像之间的差异,以及所述样本图像的融合卷积特征与所述样本图像的标签真值图像之间的差异,对所述初始血管分割模型进行参数迭代,得到所述血管分割模型。
[0216]
图7示例了一种电子设备的实体结构示意图,如图7所示,该电子设备可以包括:处理器(processor)710、通信接口(communications interface)720、存储器(memory)730和通信总线740,其中,处理器710,通信接口720,存储器730通过通信总线740完成相互间的通
信。处理器710可以调用存储器730中的逻辑指令,以执行用于多分段血管实时自动分割方法,该方法包括:获取待分割图像;将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
[0217]
此外,上述的存储器730中的逻辑指令可以通过软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本发明的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:u盘、移动硬盘、只读存储器(rom,read-only memory)、随机存取存储器(ram,random access memory)、磁碟或者光盘等各种可以存储程序代码的介质。
[0218]
另一方面,本发明还提供一种计算机程序产品,所述计算机程序产品包括计算机程序,计算机程序可存储在非暂态计算机可读存储介质上,所述计算机程序被处理器执行时,计算机能够执行上述各方法所提供的用于多分段血管实时自动分割方法,该方法包括:获取待分割图像;将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
[0219]
又一方面,本发明还提供一种非暂态计算机可读存储介质,其上存储有计算机程序,该计算机程序被处理器执行时实现以执行上述各方法提供的用于多分段血管实时自动分割方法,该方法包括:获取待分割图像;将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。
[0220]
以上所描述的装置实施例仅仅是示意性的,其中所述作为分离部件说明的单元可
以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。本领域普通技术人员在不付出创造性的劳动的情况下,即可以理解并实施。
[0221]
通过以上的实施方式的描述,本领域的技术人员可以清楚地了解到各实施方式可借助软件加必需的通用硬件平台的方式来实现,当然也可以通过硬件。基于这样的理解,上述技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品可以存储在计算机可读存储介质中,如rom/ram、磁碟、光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行各个实施例或者实施例的某些部分所述的方法。
[0222]
最后应说明的是:以上实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的精神和范围。

技术特征:
1.一种用于多分段血管实时自动分割方法,其特征在于,包括:获取待分割图像;将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。2.根据权利要求1所述的用于多分段血管实时自动分割方法,其特征在于,所述类别先验网络包括图像特征提取网络、类别先验提取网络和融合网络;所述图像特征提取网络用于对所述图像特征进行语义特征提取,得到深层特征,所述类别先验提取网络用于基于类别先验知识,对初始血管分类特征进行血管分类,得到类别先验提取特征,所述融合网络用于对所述深层特征和所述类别先验提取特征进行融合,得到类别先验特征。3.根据权利要求1所述的用于多分段血管实时自动分割方法,其特征在于,所述主干网络包括编码部分、解码部分和若干全局先验注意力模块;所述编码部分与所述解码部分连接,所述编码部分包括多个串联的编码器,所述解码部分包括多个串联的解码器;所述全局先验注意力模块用于对所述编码器输出的图像特征进行全局先验注意力提取,得到全局先验注意力特征;所述解码器用于对前一解码器输出的前一解码结果和对应的全局先验注意力模块输出的全局先验注意力特征进行解码,得到当前解码结果。4.根据权利要求3所述的用于多分段血管实时自动分割方法,其特征在于,所述全局先验注意力模块包括第一分支和第二分支;所述第一分支用于提取所述图像特征的隐藏层特征和各类别血管的语义特征,并基于所述隐藏层特征和所述各类别血管的语义特征提取第一注意力特征,并基于所述第一注意力特征和所述各类别血管的语义特征,得到第二注意力特征;所述第二分支用于基于所述第二注意力特征和所述图像特征确定所述全局先验注意力特征。5.根据权利要求1所述的用于多分段血管实时自动分割方法,其特征在于,所述结构先验网络包括稀疏掩码分支和卷积分支;所述稀疏掩码分支用于对所述特征图进行卷积操作和聚合操作,得到稀疏掩码特征,所述卷积分支用于对所述特征图进行卷积操作,得到卷积特征,并基于所述卷积特征和所述稀疏掩码特征确定所述结构先验特征。6.根据权利要求1至5中任一项所述的用于多分段血管实时自动分割方法,其特征在
于,所述血管分割模型的训练步骤包括:获取初始血管分割模型、样本图像以及所述样本图像的标签真值图像;将所述样本图像输入至初始血管分割模型,得到所述初始血管分割模型输出的血管预测分割结果、所述样本图像的图像特征、所述样本图像的稀疏掩码特征,并基于所述初始血管分割模型的主干网络中的融合卷积模块,对所述样本图像的图像特征进行融合卷积,得到融合卷积特征;基于所述血管预测分割结果与所述样本图像的标签真值图像之间的差异、所述样本图像的稀疏掩码特征与所述样本图像的标签真值图像之间的差异,以及所述样本图像的融合卷积特征与所述样本图像的标签真值图像之间的差异,对所述初始血管分割模型进行参数迭代,得到所述血管分割模型。7.一种用于多分段血管实时自动分割装置,其特征在于,包括:获取单元,用于获取待分割图像;血管分割单元,用于将所述待分割图像输入至血管分割模型,得到所述血管分割模型输出的血管分割结果;所述血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;所述主干网络用于提取所述待分割图像的图像特征,并对所述图像特征进行解码得到特征图;所述类别先验网络用于基于类别先验知识,对所述图像特征进行血管分类,得到类别先验特征,所述结构先验网络用于基于结构先验知识,对所述特征图进行血管分类,得到结构先验特征,所述动态控制网络用于对所述类别先验特征、所述结构先验特征和所述图像特征进行特征融合,得到融合特征;所述主干网络还用于基于所述融合特征和所述特征图进行血管分割。8.一种电子设备,包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现如权利要求1至6任一项所述用于多分段血管实时自动分割方法。9.一种非暂态计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至6任一项所述用于多分段血管实时自动分割方法。10.一种计算机程序产品,包括计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至6任一项所述用于多分段血管实时自动分割方法。

技术总结
本发明提供一种用于多分段血管实时自动分割方法、装置、设备及介质,其中方法包括:获取待分割图像;将待分割图像输入至血管分割模型,得到血管分割模型输出的血管分割结果;血管分割模型包括主干网络、类别先验网络、结构先验网络和动态控制网络;主干网络用于提取待分割图像的图像特征,并对图像特征进行解码得到特征图,还用于基于融合特征和特征图进行血管分割;类别先验网络用于基于类别先验知识,得到类别先验特征,结构先验网络用于基于结构先验知识,得到结构先验特征,动态控制网络用于对类别先验特征、结构先验特征和图像特征进行特征融合,得到融合特征。本发明提供的方法、装置、设备及介质,提高了血管分割的准确性和可靠性。可靠性。可靠性。


技术研发人员:刘市祺 宋猛 谢晓亮 周小虎 侯增广 马西瑶 张林森 刘暴 来志超 王超楠
受保护的技术使用者:中国科学院自动化研究所
技术研发日:2023.04.23
技术公布日:2023/8/22
版权声明

本文仅代表作者观点,不代表航空之家立场。
本文系作者授权航家号发表,未经原创作者书面授权,任何单位或个人不得引用、复制、转载、摘编、链接或以其他任何方式复制发表。任何单位或个人在获得书面授权使用航空之家内容时,须注明作者及来源 “航空之家”。如非法使用航空之家的部分或全部内容的,航空之家将依法追究其法律责任。(航空之家官方QQ:2926969996)

飞行汽车 https://www.autovtol.com/

分享:

扫一扫在手机阅读、分享本文

相关推荐