CN114332463A - 一种mr脑肿瘤图像实例分割方法、装置、设备及存储介质 - Google Patents
一种mr脑肿瘤图像实例分割方法、装置、设备及存储介质 Download PDFInfo
- Publication number
- CN114332463A CN114332463A CN202111671947.1A CN202111671947A CN114332463A CN 114332463 A CN114332463 A CN 114332463A CN 202111671947 A CN202111671947 A CN 202111671947A CN 114332463 A CN114332463 A CN 114332463A
- Authority
- CN
- China
- Prior art keywords
- network
- feature map
- brain tumor
- convolution
- tumor image
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 230000011218 segmentation Effects 0.000 title claims abstract description 134
- 208000003174 Brain Neoplasms Diseases 0.000 title claims abstract description 133
- 238000000034 method Methods 0.000 title claims abstract description 72
- 238000003860 storage Methods 0.000 title claims abstract description 14
- 238000010586 diagram Methods 0.000 claims abstract description 60
- 238000001514 detection method Methods 0.000 claims abstract description 36
- 206010028980 Neoplasm Diseases 0.000 claims abstract description 32
- 238000005070 sampling Methods 0.000 claims abstract description 25
- 238000012545 processing Methods 0.000 claims description 66
- 238000012549 training Methods 0.000 claims description 39
- 238000011176 pooling Methods 0.000 claims description 23
- 230000008569 process Effects 0.000 claims description 22
- 238000010606 normalization Methods 0.000 claims description 18
- 230000006870 function Effects 0.000 claims description 16
- 230000004927 fusion Effects 0.000 claims description 13
- 230000009467 reduction Effects 0.000 claims description 9
- 238000013528 artificial neural network Methods 0.000 claims description 8
- 238000004590 computer program Methods 0.000 claims description 8
- 238000007781 pre-processing Methods 0.000 claims description 8
- 210000002569 neuron Anatomy 0.000 claims description 6
- 238000004891 communication Methods 0.000 claims description 5
- 230000003321 amplification Effects 0.000 claims description 4
- 238000003199 nucleic acid amplification method Methods 0.000 claims description 4
- 230000004913 activation Effects 0.000 claims description 3
- JXSJBGJIGXNWCI-UHFFFAOYSA-N diethyl 2-[(dimethoxyphosphorothioyl)thio]succinate Chemical compound CCOC(=O)CC(SP(=S)(OC)OC)C(=O)OCC JXSJBGJIGXNWCI-UHFFFAOYSA-N 0.000 claims description 3
- 230000005489 elastic deformation Effects 0.000 claims description 3
- 230000008859 change Effects 0.000 abstract description 8
- 230000000694 effects Effects 0.000 abstract description 8
- 230000007246 mechanism Effects 0.000 abstract description 6
- 230000003416 augmentation Effects 0.000 abstract description 3
- 238000012216 screening Methods 0.000 abstract description 3
- 238000013527 convolutional neural network Methods 0.000 description 12
- 238000013461 design Methods 0.000 description 11
- 238000005457 optimization Methods 0.000 description 8
- 238000004422 calculation algorithm Methods 0.000 description 5
- 238000000605 extraction Methods 0.000 description 5
- 238000003709 image segmentation Methods 0.000 description 5
- 238000002595 magnetic resonance imaging Methods 0.000 description 5
- 210000001519 tissue Anatomy 0.000 description 5
- 230000008901 benefit Effects 0.000 description 4
- 238000004458 analytical method Methods 0.000 description 3
- 150000001875 compounds Chemical class 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 238000003384 imaging method Methods 0.000 description 3
- 238000012360 testing method Methods 0.000 description 3
- 230000009286 beneficial effect Effects 0.000 description 2
- 210000004556 brain Anatomy 0.000 description 2
- 230000003902 lesion Effects 0.000 description 2
- 230000007935 neutral effect Effects 0.000 description 2
- 210000000056 organ Anatomy 0.000 description 2
- 238000005096 rolling process Methods 0.000 description 2
- 208000014644 Brain disease Diseases 0.000 description 1
- 101100512352 Enterococcus faecalis (strain TX4000 / JH2-2) mapP gene Proteins 0.000 description 1
- 238000005481 NMR spectroscopy Methods 0.000 description 1
- 206010030113 Oedema Diseases 0.000 description 1
- RTAQQCXQSZGOHL-UHFFFAOYSA-N Titanium Chemical compound [Ti] RTAQQCXQSZGOHL-UHFFFAOYSA-N 0.000 description 1
- 230000003044 adaptive effect Effects 0.000 description 1
- 230000002457 bidirectional effect Effects 0.000 description 1
- 201000007983 brain glioma Diseases 0.000 description 1
- 210000005013 brain tissue Anatomy 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 210000003710 cerebral cortex Anatomy 0.000 description 1
- 210000001175 cerebrospinal fluid Anatomy 0.000 description 1
- 238000012512 characterization method Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 238000003745 diagnosis Methods 0.000 description 1
- 201000010099 disease Diseases 0.000 description 1
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 description 1
- 238000007667 floating Methods 0.000 description 1
- 238000009472 formulation Methods 0.000 description 1
- 210000004884 grey matter Anatomy 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 238000003475 lamination Methods 0.000 description 1
- 239000000203 mixture Substances 0.000 description 1
- 230000001338 necrotic effect Effects 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 231100000915 pathological change Toxicity 0.000 description 1
- 230000036285 pathological change Effects 0.000 description 1
- 230000001575 pathological effect Effects 0.000 description 1
- 230000007310 pathophysiology Effects 0.000 description 1
- 238000004393 prognosis Methods 0.000 description 1
- 230000001737 promoting effect Effects 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 210000003625 skull Anatomy 0.000 description 1
- 210000004872 soft tissue Anatomy 0.000 description 1
- 238000003325 tomography Methods 0.000 description 1
- 239000013598 vector Substances 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
- 210000004885 white matter Anatomy 0.000 description 1
Images
Landscapes
- Image Analysis (AREA)
- Magnetic Resonance Imaging Apparatus (AREA)
Abstract
本发明涉及计算机视觉技术领域,公开了一种MR脑肿瘤图像实例分割方法、装置、设备及存储介质,即在现有Mask RCNN实例分割模型和特征金字塔网络FPN的基础上,在FPN上增加了一条自底而上路径增强Bottom‑up Path Augmentation,然后在FPN的自顶而下通路Top‑down pathway的上采样路径和Bottom‑up Path Augmentation的下采样路径中分别引入卷积层注意力机制CBAM,可以更好改善肿瘤形状大小因为个体差异和时间变化而产生的改变问题,同时融合低分辨率特征图和高分辨率特征图,可以有效提高小目标的检测效果,降低小尺寸肿瘤的漏检率,为肿瘤的早筛提供帮助,如此不但在分割性能上有提升且具有较好的通用性。
Description
技术领域
本发明属于计算机视觉技术领域,具体地涉及一种MR脑肿瘤图像实例分割方法、装置、设备及存储介质。
背景技术
磁共振成像(MRI,Magnetic Resonance Imaging)是一种非侵入式成像技术,允许对器官或肿瘤的介观特征进行无创性表征。MRI技术可提供良好的空间解剖信息,具有良好的软组织对比度检测能力,无需重建就可获得任意方位的断层扫描,所获取的多序列多模态成像可为明确病变性质提供丰富的影像信息,帮助医生和研究人员理解大脑疾病的病理生理学,辅助影像诊断、预后和肿瘤治疗。在临床环境中,病灶精准检测与分割对患者病情评估和治疗方案制定有重要意义。
然而,MR脑肿瘤图像存在如下客观因素:(1)脑部组织结构的复杂性,即MRI脑影像中既有正常组织,如脑皮层、脑脊液、白质和灰质等,也有病变组织,如增强肿瘤、非增强肿瘤、水肿以及肿瘤内部的坏死区域等,部分组织结构成像相似性高,对比度低,导致不易区分;(2)多尺度,即肿瘤形状大小各异以及在不同时期形状大小多变;(3)空间位置不固定,即肿瘤区域不固定,存在个体差异;(4)多目标,即多个目标或含有多个子结构,容易漏检;(5)类别不平衡,即病灶区域的特征信息在MR图像中占比较小。
上述因素导致面向MR脑肿瘤精准分割仍面临很大挑战,目前很多主流的分割算法功能较为单一,只能对上述因素中的某一个问题进行改善,通用性较差。如有人利用U-Net或其改进的网络结构在医学图像的不同组织器官或病灶进行分割,通过含有跳跃连接的编码—解码结构的网络结构,有效地提升分割精度。针对脑胶质瘤多子区域的分割问题,还有人利用多任务网络将多子结构分割问题集成到同一个网络,并通过课程学习的方式逐次实现多任务分支的训练。还有人在保持精度的情况下通过轻量化来提高脑肿瘤的分割效率。当肿瘤相距较远时,大多数模型只能分割出其中一部分目标,缺乏全局视野,易出现漏检情况。主流模型在多尺度肿瘤检测上效果不理想,图像中尺寸较小的肿瘤存在较高的漏检率,对于尺度较大的肿瘤则难以较好的实现完整分割。由此针对MR脑肿瘤图像存在的客观复杂性,有必要设计一个能改善性能且通用性较好的分割模型及分割方案。
由于在卷积神经网络(Convolutional Neural Network,CNN)上可通过堆叠多个卷积层、池化层和全连接层以及将图像与核特征参数进行卷积,形成具有强鲁棒性和自适应性的特征学习模型,因此相对于人工分割和传统分割方法,将CNN应用在脑肿瘤分割的最大优势可在于能够提取脑部MRI图像中肿瘤复杂特征。CNN的特征检测层通过对训练数据的学习,避免了显式的特征抽取,隐式地从训练数据中进行学习,能够大幅度提高脑肿瘤图像分割精度。因此有人开始尝试把CNN卷积神经网络用于脑肿瘤分割,如Pereira等基于CNN设计了自动分割MRI脑肿瘤的方法。
进一步的,还有人在CNN技术的基础上提出了RCNN神经网络,即将卷积神经网络应用于目标检测,利用选择搜索(Selective Search)区域推荐方法从图像中提取推荐区域(Region Proposals,RP),实现图像特征自动提取,最终进行目标分类和目标定位。然而由于每个推荐区域都需要经过CNN网络进行提取特征,造成大量的特征文件存储空间和运行时间的消耗。为此又有人做了改进,提出SPP-NET,即在特征提取阶段,在全连接层前面加入一个空间金字塔结构,适应各种不同尺寸的特征图;通过空间金字塔池化提取固定的特征向量,对整张图进行一次特征提取,提高了运算速度且减少了存储空间,并首次提出了共享卷积。但是这种方法也存在如下缺点:分类器使用SVM,不能端到端训练;进行分阶段训练,训练步骤复杂,为此又有人使用softmax去代替SVM进行多分类的预测,从而实现了端到端的训练。但是这种新改进方法也有缺点:推荐区域的提取还是通过选择搜索,并且只能在CPU中运行这个算法,所以这个阶段浪费了大量时间。
为了解决推荐区域生成步骤消耗大量计算资源,导致检测速度过慢的问题,又有人提出Faster RCNN,即用RPN((Region Proposal Network)网络代替选择搜索去生成推荐区域,该区域推荐网络与检测网络共享整张图的卷积特征,使得区域推荐基本不消耗任何计算时间。然而Faster RCNN存在的问题是当下采样Conv4层的输出到达RPN层后,RPN使用单个高层特征图(single feature map)进行物体的分类和边框bounding box(简写bbox)的回归。如果只使用一个尺度的特征图,可能这层特征的语义表达不强,但如果选择后面层,虽然语义表达强了,但其位置信息可能就不够充分了,小物体本身具有的像素信息较少,在下采样的过程中很容易被丢失。
为了处理这种物体大小差异明显的检测问题,经典的方法是利用图像金字塔的方式进行多尺度变化增强,但这样会带来极大的计算量。为此还有人提出了特征金字塔的网络结构(Feature Pyramid Networks,FPN),其以特征金字塔为基础结构,对每一层级的特征图分别进行预测,解决目标检测在处理多尺度变化问题,同时付出较小的时间代价。同时还有人开发出集目标检测(Object detection)和实例分割(Instance Segmentation)为一体的网络模型——Mask RCNN。可在Faster-RCNN的基础上,加入了全连接的分割子网,可在原目标检测任务基础上增加分割任务。Mask RCNN即可以定位小目标也能对目标进行分割,本质上是一个实例分割,实例分割不仅要做像素级别的分类,还要将同一类别不同个体分出来。相较于语义分割(SemanticSegmentation),对分割算法提出了更高的要求,实例分割对同类物体有着更为精细的分割。然而,由于最高层级特征和较低层级特征之间的路径很长,增加了Mask RCNN访问准确定位信息的难度。
目前,两阶段检测和实例分割模型,如Mask RCNN,由三个部分组成:骨干backbone网络、颈部neck网络和头部head网络,其中,neck网络的经典优化方法有FPN、PANet和BiFPN等。考虑网络的低层特征,由于采样次数较少,含有更多的目标位置信息,但语义信息少,不利于目标识别;高层特征中含有更多的语义信息,但目标定位不准确。为了解决这些问题,Mask RCNN的neck采用特征金字塔网络FPN,即通过Bottom-UP方式增加融合高低层特征使网络适应不同尺度的图片,对多尺度的特征图进行独立预测,提高小尺寸目标的检测和分割效果;为此有人基于FPN增加了自底而上路径增强Bottom-up Path Augmentation,可在分割中增加低层信息的利用率;也有人在neck网络中采用加权双向特征金字塔网络BiFPN级联,该BiFPN与PANet相比,增加了跨层链接,可对图像特征进行深度融合,提高分割精度,但同时降低了分割效率。
因此上述FPN、PANet和BiFPN三种优化方式各有优缺点,其中,FPN优化方式虽然分割效率高,但是分割精度不够;PANet优化方式是在分割精度和效率上寻找平衡;BiFPN优化方式虽然分割精度高,但分割效率低。由此如何基于Mask RCNN网络模型,提出一种具有更高分割精度和效率的MR脑肿瘤图像实例分割方案,是本领域技术人员亟需研究的课题。
发明内容
为了解决现有MR脑肿瘤实例分割技术所存在分割精度和效率有限的问题,本发明目的在于提供一种新型的MR脑肿瘤图像实例分割方法、装置、计算机设备及计算机可读存储介质。
第一方面,本发明提供了一种MR脑肿瘤图像实例分割方法,包括:
获取脑肿瘤公开数据训练集,其中,所述脑肿瘤公开数据训练集包含有多个MR脑肿瘤图像样本;
针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像;
将所有的所述样本图像送入集目标检测和实例分割于一体的Mask RCNN网络模型进行训练,得到训练完毕的MR脑肿瘤图像实例分割模型,其中,所述Mask RCNN网络模型由骨干backbone网络、颈部neck网络和头部head网络构成,所述backbone网络采用残差网络,所述neck网络采用特征金字塔网络FPN,所述head网络包括有预测框分类网络、边框bbox回归网络和mask分支网络;
所述残差网络用于在输入样本图像后,输出与该样本图像对应的六个第一特征图;
所述特征金字塔网络FPN的自顶而下通路Top-down pathway设计为:在从第三预测层P3至第七预测层P7的各个预测层中分别引入第一卷积注意力模块,以便采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,其中,i为自然数且有i∈[3,7];
所述残差网络还用于经过不同的降采样次数,使得所述第一细化新特征图Fi″在经过1*1降维处理和2倍上采样处理后,与由下一预测层输出的且经过1*1降维处理的另一第一细化新特征图F″i-1进行单位加融合,然后通过对单位加融合结果进行3*3卷积处理,得到第二特征图P″i-1;
在所述特征金字塔网络FPN中增加自底而上路径增强Bottom-upPathAugmentation,并设计为:先在所述各个预测层中分别引入第二卷积注意力模块,以便也采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图Ni″,再自底而上地进行如下迭代处理得到用于预测框分类、边框bbox回归和mask分支生成的新第二细化新特征图:对低层的所述第二细化新特征图Ni″进行一个步长为2的3*3卷积处理和下采样处理,然后将处理结果与横向连接传递过来的且高层的第二特征图P″i+1进行单位加融合,最后通过对单位加融合结果进行3*3卷积处理,得到新第二细化新特征图N″i+1;
将待处理MR图像导入所述MR脑肿瘤图像实例分割模型,输出实例分割结果。
基于上述发明内容,提供了一种基于注意力机制的MR脑肿瘤实例分割方案,即在现有Mask RCNN实例分割模型和特征金字塔网络FPN的基础上,在FPN上增加了一条自底而上路径增强Bottom-up Path Augmentation,然后在FPN的自顶而下通路Top-down pathway的上采样路径和Bottom-up Path Augmentation的下采样路径中分别引入卷积层注意力机制CBAM,可以更好改善肿瘤形状大小因为个体差异和时间变化而产生的改变问题,同时融合低分辨率特征图和高分辨率特征图,可以有效提高小目标的检测效果,降低小尺寸肿瘤的漏检率,为肿瘤的早筛提供帮助,以及由于是在FPN基础上增加自底向上的增强路径,可以利用低层特征的精准定位信息,以及由于是在增强路径上再引入注意力机制CBAM,可形成对肿瘤特征图的更优关注系数,从而更好地保持肿瘤在高维非线性空间的特征信息,使得对目标的空间位置更敏感,提高多目标检测和分割的有效性。同时本申请方案可兼顾分割效率和分割精度,并且通过实验分析证明在分割精度上高于PANet,而与三级级联BiFPN相比,虽然在精度方面持平,但在效率上比BiFPN更高,时间代价更少,进而不但在分割性能上有提升且具有较好的通用性。此外,由于CBAM是轻量级的通用模块,因此可以忽略该模块的开销,使得分割效率所受影响不大,进而确保了在模型性能提升的同时,保持了算法的高效率;以及由于是集目标检测和实例分割为一体,并且BBox对区域划分和类别判断较为准确,使得输出类别的概率对临床意义有较高的参考价值,因此相对于纯语义分割,更有利于处理多目标和小目标的检测与分割,也使得MR脑肿瘤的实例分割模型具有良好的通用性。
在一个可能的设计中,所述脑肿瘤公开数据训练集采用脑肿瘤公开数据集BRATS2018中的训练子集,其中,所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本分别有如下4种模态:T1模态、T2模态、T1ec模态和Flair模态。
在一个可能的设计中,针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像,包括:将MR脑肿瘤图像样本标准化,使其具有零均值与单位标准差。
在一个可能的设计中,针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像,还包括:
对标准化后的MR脑肿瘤图像样本进行如下方式的数据增扩处理:在各个通道上进行随机强度偏移处理,以及在三轴数据上进行随机旋转处理、镜像翻转处理、弹性形变处理和/或缩放处理。
在一个可能的设计中,所述残差网络采用包含有七组卷积层的ResNet101网络结构;
在所述ResNet101网络结构中,第一组卷积层的大小是64*7*7且后接最大值池化层;
在所述ResNet101网络结构中,第二组卷积层包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为64*1*1、64*3*3和256*1*1;
在所述ResNet101网络结构中,第三组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为128*1*1、128*3*3和512*1*1;
在所述ResNet101网络结构中,第四组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为256*1*1、256*3*3和1024*1*1;
在所述ResNet101网络结构中,第五组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为512*1*1、512*3*3和2048*1*1;
在所述ResNet101网络结构中,第六组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为1024*1*1、1024*3*3和4096*1*1;
在所述ResNet101网络结构中,第七组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为2048*1*1、2048*3*3和8192*1*1。
在一个可能的设计中,在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,包括:
式中,σ()表示归一化函数sigmoid()且用于根据神经网络的传播回馈不断更新权重系数以便指导模型对来自低层的肿瘤特征进行选择,AvgPool()表示平均池化函数,MaxPool()表示最大池化函数,MLP()表示多层感知器,表示对所述第一特征图Fi进行平均池化后得到的特征图,表示对所述第一特征图Fi进行最大池化后得到的特征图,W0和W1表示所述多层感知器MLP()的权重且有W0∈RC/r×C和W1∈RC×C/r,R表示实数,C/r表示所述多层感知器MLP()的第一层神经元个数,r表示预设减少率,C表示所述多层感知器MLP()的第二层神经元个数,所述多层感知器MLP()的激活函数采用线性整流函数Relu;
在一个可能的设计中,在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图Ni″,包括:
第二方面,本发明提供了一种MR脑肿瘤图像实例分割装置,包括有依次通信连接的训练集获取单元、预处理单元、模型训练单元和模型应用单元;
所述训练集获取单元,用于获取脑肿瘤公开数据训练集,其中,所述脑肿瘤公开数据训练集包含有多个MR脑肿瘤图像样本;
所述预处理单元,用于针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像;
所述模型训练单元,用于将所有的所述样本图像送入集目标检测和实例分割于一体的Mask RCNN网络模型进行训练,得到训练完毕的MR脑肿瘤图像实例分割模型,其中,所述Mask RCNN网络模型由骨干backbone网络、颈部neck网络和头部head网络构成,所述backbone网络采用残差网络,所述neck网络采用特征金字塔网络FPN,所述head网络包括有预测框分类网络、边框bbox回归网络和mask分支网络;
所述残差网络用于在输入样本图像后,输出与该样本图像对应的六个第一特征图;
所述特征金字塔网络FPN的自顶而下通路Top-down pathway设计为:在从第三预测层P3至第七预测层P7的各个预测层中分别引入第一卷积注意力模块,以便采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,其中,i为自然数且有i∈[3,7];
所述残差网络还用于经过不同的降采样次数,使得所述第一细化新特征图Fi″在经过1*1降维处理和2倍上采样处理后,与由下一预测层输出的且经过1*1降维处理的另一第一细化新特征图F″i-1进行单位加融合,然后通过对单位加融合结果进行3*3卷积处理,得到第二特征图P″i-1;
在所述特征金字塔网络FPN中增加自底而上路径增强Bottom-up PathAugmentation,并设计为:先在所述各个预测层中分别引入第二卷积注意力模块,以便也采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入第二特征图Pi-1后,得到与该第二特征图P″i-1对应的第二细化新特征图Ni″,再自底而上地进行如下迭代处理得到用于预测框分类、边框bbox回归和mask分支生成的新第二细化新特征图:对低层的所述第二细化新特征图Ni″进行一个步长为2的3*3卷积处理和下采样处理,然后将处理结果与横向连接传递过来的且高层的第二特征图P″i+1进行单位加融合,最后通过对单位加融合结果进行3*3卷积处理,得到新第二细化新特征图N″i+1;
所述模型应用单元,用于将待处理MR图像导入所述MR脑肿瘤图像实例分割模型,输出实例分割结果。
第三方面,本发明提供了一种计算机设备,包括有依次通信连接的存储器、处理器和收发器,其中,所述存储器用于存储计算机程序,所述收发器用于收发信息,所述处理器用于读取所述计算机程序,执行如第一方面或第一方面中任意可能设计所述的MR脑肿瘤图像实例分割方法。
第四方面,本发明提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有指令,当所述指令在计算机上运行时,执行如上第一方面或第一方面中任意可能设计所述的MR脑肿瘤图像实例分割方法。
第五方面,本发明提供了一种包含指令的计算机程序产品,当所述指令在计算机上运行时,使所述计算机执行如上第一方面或第一方面中任意可能设计所述的MR脑肿瘤图像实例分割方法。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
图1是本发明提供的MR脑肿瘤图像实例分割方法的流程示意图。
图2是本发明提供的卷积注意力模块CBAM的工作过程示意图。
图3是本发明提供的CBAM-FPN模型的部分网络结构示意图。
图4是本发明提供的MR脑肿瘤图像实例分割装置的结构示意图。
图5是本发明提供的计算机设备的结构示意图。
具体实施方式
下面结合附图及具体实施例来对本发明作进一步阐述。在此需要说明的是,对于这些实施例方式的说明虽然是用于帮助理解本发明,但并不构成对本发明的限定。本文公开的特定结构和功能细节仅用于描述本发明示例的实施例。然而,可用很多备选的形式来体现本发明,并且不应当理解为本发明限制在本文阐述的实施例中。
应当理解,尽管本文可能使用术语第一和第二等等来描述各种对象,但是这些对象不应当受到这些术语的限制。这些术语仅用于区分一个对象和另一个对象。例如可以将第一对象称作第二对象,并且类似地可以将第二对象称作第一对象,同时不脱离本发明的示例实施例的范围。
应当理解,对于本文中可能出现的术语“和/或”,其仅仅是一种描述关联对象的关联关系,表示可以存在三种关系,例如,A和/或B,可以表示:单独存在A、单独存在B或者同时存在A和B等三种情况;对于本文中可能出现的术语“/和”,其是描述另一种关联对象关系,表示可以存在两种关系,例如,A/和B,可以表示:单独存在A或者同时存在A和B等两种情况;另外,对于本文中可能出现的字符“/”,一般表示前后关联对象是一种“或”关系。
如图1~3所示,本实施例第一方面提供的所述MR脑肿瘤图像实例分割方法,可以但不限于由具有一定计算资源的计算机设备执行,例如由个人计算机(PersonalComputer,PC,指一种大小、价格和性能适用于个人使用的多用途计算机;台式机、笔记本电脑到小型笔记本电脑和平板电脑以及超级本等都属于个人计算机)、智能手机、个人数字助理(Personal digital assistant,PAD)或可穿戴设备等电子设备执行。如图1所示,所述MR脑肿瘤图像实例分割方法,可以但不限于包括有如下步骤S1~S4。
S1.获取脑肿瘤公开数据训练集,其中,所述脑肿瘤公开数据训练集包含有多个MR脑肿瘤图像样本。
在所述步骤S1中,具体的,所述脑肿瘤公开数据训练集可采用脑肿瘤公开数据集BRATS2018中的训练子集,其中,所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本分别有如下4种模态:T1模态、T2模态、T1ec模态和Flair模态。前述四种模态可以理解为核磁共振图像的四个不同纬度信息,每个序列的图像形状分别为(155,240,240)。此外,所述脑肿瘤公开数据集BRATS2018具体共有285个病例,并具有三个子集,即训练子集、测试子集和排行榜子集。
S2.针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像。
在所述步骤S2中,虽然所述脑肿瘤公开数据集BRATS2018已经对配准,并且重采样到1*1*1mm以及实现头骨剥离,但是对于分割任务来说,还是需要做灰度数据归一化处理,以便送入后续模型进行模型训练,即具体的,针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像,包括:将MR脑肿瘤图像样本标准化,使其具有零均值与单位标准差,其中,前述标准化的处理方式为常规预处理方式。此外,为了提升后续已训练模型的目标检测及分割能力,还需要对标准化后的MR脑肿瘤图像样本进行数据增扩处理,即针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像,还包括:对标准化后的MR脑肿瘤图像样本进行如下方式的数据增扩处理:在各个通道上进行随机强度偏移处理,以及在三轴数据上进行随机旋转处理、镜像翻转处理、弹性形变处理和/或缩放处理,其中,前述几种处理方式同样为常规的预处理方式。
S3.将所有的所述样本图像送入集目标检测和实例分割于一体的Mask RCNN网络模型进行训练,得到训练完毕的MR脑肿瘤图像实例分割模型,其中,所述Mask RCNN网络模型由骨干backbone网络、颈部neck网络和头部head网络构成,所述backbone网络采用残差网络,所述neck网络采用特征金字塔网络FPN,所述head网络包括有预测框分类网络、边框bbox回归网络和mask分支网络。
在所述步骤S3中,所述Mask RCNN网络模型为现有网络模型,具有如下基本结构:backbone网络、neck网络和head网络,其中,所述残差网络用于在输入样本图像后,输出与该样本图像对应的六个第一特征图,详细可采用包含有七组卷积层的ResNet101网络结构;在所述ResNet101网络结构中,第一组卷积层的大小是64*7*7且后接最大值池化层;在所述ResNet101网络结构中,第二组卷积层包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为64*1*1、64*3*3和256*1*1;在所述ResNet101网络结构中,第三组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为128*1*1、128*3*3和512*1*1;在所述ResNet101网络结构中,第四组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为256*1*1、256*3*3和1024*1*1;在所述ResNet101网络结构中,第五组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为512*1*1、512*3*3和2048*1*1;在所述ResNet101网络结构中,第六组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为1024*1*1、1024*3*3和4096*1*1;在所述ResNet101网络结构中,第七组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为2048*1*1、2048*3*3和8192*1*1。此外,所述特征金字塔网络FPN已在背景技术中有所介绍,所述预测框分类网络、所述边框bbox回归网络和所述mask分支网络均为现有Mask RCNN网络模型中head网络的常规配置。
本实施例主要针对所述neck网络进行了优化设计,即先将所述特征金字塔网络FPN的自顶而下通路Top-down pathway设计为:在从第三预测层P3至第七预测层P7的各个预测层中分别引入第一卷积注意力模块,以便采用先通道后空间的策略,在空间(Spatial)和通道(Channel)两个维度上进行重要特征加权,使得在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,其中,i为自然数且有i∈[3,7]。所述自顶而下通路Top-down pathway是一个高层特征图上采样过程(每层之间上采样比例为2);从所述第三预测层P3至所述第七预测层P7的网络结构为所述特征金字塔网络FPN中的常规配置。所述第一卷积注意力模块是一种用于前馈卷积神经网络的且简单而有效的注意力模块,可在给定一个中间特征图后,沿着两个独立的维度(即通道和空间)依次推断注意力图,然后将注意力图与输入特征图相乘以进行自适应特征优化;由于CBAM(Convolutional Block Attention Module,卷积注意力模块)是轻量级的通用模块,因此可以忽略该模块的开销而将其无缝集成到任何CNN架构中,并且可以与基础CNN一起进行端到端训练。
具体的,如图2所示,在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,包括但不限于有如下步骤S311~S314。
式中,σ()表示归一化函数sigmoid()且用于根据神经网络的传播回馈不断更新权重系数以便指导模型对来自低层的肿瘤特征进行选择,AvgPool()表示平均池化函数,MaxPool()表示最大池化函数,MLP()表示多层感知器,表示对所述第一特征图Fi进行平均池化后得到的特征图,表示对所述第一特征图Fi进行最大池化后得到的特征图,W0和W1表示所述多层感知器MLP()的权重且有W0∈RC/r×C和W1∈RC×C/r,R表示实数,C/r表示所述多层感知器MLP()的第一层神经元个数,r表示预设减少率,C表示所述多层感知器MLP()的第二层神经元个数,所述多层感知器MLP()的激活函数采用线性整流函数Relu。
在得到所述第一细化新特征图Fi″后,所述残差网络还将用于经过不同的降采样次数,使得所述第一细化新特征图Fi″在经过1*1降维处理和2倍上采样处理后,与由下一预测层输出的且经过1*1降维处理的另一第一细化新特征图F″i-1进行单位加融合,然后通过对单位加融合结果进行3*3卷积处理,得到第二特征图P″i-1。前述过程的具体公式可如下所示:
本实施例针对所述neck网络进行的优化设计,还有在所述特征金字塔网络FPN中增加自底而上路径增强Bottom-up Path Augmentation,并设计为:先在所述各个预测层中分别引入第二卷积注意力模块,以便也采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图Ni″,再自底而上地进行如下迭代处理得到用于预测框分类、边框bbox回归和mask分支生成的新第二细化新特征图:对低层的所述第二细化新特征图Ni″进行一个步长为2的3*3卷积处理和下采样处理,然后将处理结果与横向连接传递过来的且高层的第二特征图P″i+1进行单位加融合,最后通过对单位加融合结果进行3*3卷积处理,得到新第二细化新特征图N″i+1。
所述第二卷积注意力模块也是一种用于前馈卷积神经网络的且简单而有效的注意力模块,可参照前述步骤S311~S314,在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图Ni″,即具体的,包括有如下步骤S321~S324:S321.将第二特征图P″i-1送入所述第二卷积注意力模块中的通道注意力子模块,以便利用学习方式获取在通道注意力维度上的第三权重系数S322.对所述第三权重系数和所述第二特征图P″i-1进行单位乘处理,得到第二新特征图P′i-1;S323.将所述第二新特征图P′i-1送入所述第二卷积注意力模块中的空间注意力子模块,以便获取在空间注意力维度上的第四权重系数S324.对所述第四权重系数和所述第二新特征图P′i-1进行卷积操作处理,得到与所述第二特征图P″i-1对应的第二细化新特征图Ni″。前述步骤S321~S324的具体细节可参见前述步骤S311~S314,于此不再赘述。此外,如图3所示,针对得到新第二细化新特征图N″i+1的具体过程,可用如下迭代公式表示:
在将所有的所述样本图像送入上述改进的Mask RCNN网络模型中后,即可对该Mask RCNN网络模型进行训练,得到能够实现集目标检测和实例分割于一体的MR脑肿瘤图像实例分割模型。此外,为了验证该MR脑肿瘤图像实例分割模型,还可以针对所述脑肿瘤公开数据集BRATS2018中的测试子集,在图像样本进行灰度数据归一化处理后送入所述MR脑肿瘤图像实例分割模型中进行实例分割正确性验证。
示例的,本实施例还应用如下实验条件:(1)硬件环境:CPU:Intel(R)Xeon(R)CPUE5-2660 [email protected];内存:24GB;显卡:TITAN RTX 24GB Ram;(2)软件环境:OS:Ubuntu16.04.6 LTS;开发软件:python;基础开发框架:PyTorch,对本实施例提出的优化模型(本申请命名为CBAM-FPN模型)和现有几种模型(即Mask RCNN、PANet和BIFPN等)进行了对比试验,即所有比较模型的backbone网络均采用ResNet101网络,head网络均采用与Mask RCNN相同的head,精度指标采用Dice(Dice Similarity coefficient,骰子相似系数)和AP(Average Precision,平均精度),效率指标采用参数量和FLOPs(即“每秒所执行的浮点运算次数”Floating-point operations per second的缩写),并依据BRATS2018脑肿瘤结构的划分,从三个肿瘤区域:全肿瘤区域(WT)、肿瘤核心区域(TC)和肿瘤增强区域(ET)等评价分割性能,最终有如下表1~表3所示的实验结果及分析。
表1.脑肿瘤分割上的性能指标对比
表2.不同尺寸脑肿瘤检测效果对比
表3.不同分割模型参数量和计算力对比
模型 | 参数量(M) | FLOPs(B) |
MaskRCNN | 43.21 | 126.33 |
PANet | 52.37 | 162.92 |
BIFPN(3次级联) | 67.02 | 262.31 |
CBAM-FPN | 54.80 | 173.75 |
从上述表2可以看出,本实施例所提出的CBAM-FPN模型在小目标检测上有一定的优势,并从上述表3可以看出,在相同backbone网络和head网络情况下,所述CBAM-FPN模型的参数量介于PANet和BIFPN(3次级联)之间,虽然CBAM-FPN模型的运算时间与PANet基本持平,但比BIFPN(3次级联)短。
S4.将待处理MR图像导入所述MR脑肿瘤图像实例分割模型,输出实例分割结果。
在所述步骤S4中,所述待处理MR图像同样需要在进行灰度数据归一化处理后,才送入所述MR脑肿瘤图像实例分割模型中进行实例分割,得到实例分割结果。
由此基于前述步骤S1~S4所描述的MR脑肿瘤图像实例分割方法,提供了一种基于注意力机制的MR脑肿瘤实例分割方案,即在现有Mask RCNN实例分割模型和特征金字塔网络FPN的基础上,在FPN上增加了一条自底而上路径增强Bottom-up Path Augmentation,然后在FPN的自顶而下通路Top-down pathway的上采样路径和Bottom-up PathAugmentation的下采样路径中分别引入卷积层注意力机制CBAM,可以更好改善肿瘤形状大小因为个体差异和时间变化而产生的改变问题,同时融合低分辨率特征图和高分辨率特征图,可以有效提高小目标的检测效果,降低小尺寸肿瘤的漏检率,为肿瘤的早筛提供帮助,以及由于是在FPN基础上增加自底向上的增强路径,可以利用低层特征的精准定位信息,以及由于是在增强路径上再引入注意力机制CBAM,可形成对肿瘤特征图的更优关注系数,从而更好地保持肿瘤在高维非线性空间的特征信息,使得对目标的空间位置更敏感,提高多目标检测和分割的有效性。同时本申请方案可兼顾分割效率和分割精度,并且通过实验分析证明在分割精度上高于PANet,而与三级级联BiFPN相比,虽然在精度方面持平,但在效率上比BiFPN更高,时间代价更少,进而不但在分割性能上有提升且具有较好的通用性。此外,由于CBAM是轻量级的通用模块,因此可以忽略该模块的开销,使得分割效率所受影响不大,进而确保了在模型性能提升的同时,保持了算法的高效率;以及由于是集目标检测和实例分割为一体,并且BBox对区域划分和类别判断较为准确,使得输出类别的概率对临床意义有较高的参考价值,因此相对于纯语义分割,更有利于处理多目标和小目标的检测与分割,也使得MR脑肿瘤的实例分割模型具有良好的通用性。
如图4所示,本实施例第二方面提供了一种实现第一方面所述的MR脑肿瘤图像实例分割方法的虚拟装置,包括有依次通信连接的训练集获取单元、预处理单元、模型训练单元和模型应用单元;
所述训练集获取单元,用于获取脑肿瘤公开数据训练集,其中,所述脑肿瘤公开数据训练集包含有多个MR脑肿瘤图像样本;
所述预处理单元,用于针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像;
所述模型训练单元,用于将所有的所述样本图像送入集目标检测和实例分割于一体的Mask RCNN网络模型进行训练,得到训练完毕的MR脑肿瘤图像实例分割模型,其中,所述Mask RCNN网络模型由骨干backbone网络、颈部neck网络和头部head网络构成,所述backbone网络采用残差网络,所述neck网络采用特征金字塔网络FPN,所述head网络包括有预测框分类网络、边框bbox回归网络和mask分支网络;
所述残差网络用于在输入样本图像后,输出与该样本图像对应的六个第一特征图;
所述特征金字塔网络FPN的自顶而下通路Top-down pathway设计为:在从第三预测层P3至第七预测层P7的各个预测层中分别引入第一卷积注意力模块,以便采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,其中,i为自然数且有i∈[3,7];
所述残差网络还用于经过不同的降采样次数,使得所述第一细化新特征图Fi″在经过1*1降维处理和2倍上采样处理后,与由下一预测层输出的且经过1*1降维处理的另一第一细化新特征图F″i-1进行单位加融合,然后通过对单位加融合结果进行3*3卷积处理,得到第二特征图P″i-1;
在所述特征金字塔网络FPN中增加自底而上路径增强Bottom-up PathAugmentation,并设计为:先在所述各个预测层中分别引入第二卷积注意力模块,以便也采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图Ni″,再自底而上地进行如下迭代处理得到用于预测框分类、边框bbox回归和mask分支生成的新第二细化新特征图:对低层的所述第二细化新特征图Ni″进行一个步长为2的3*3卷积处理和下采样处理,然后将处理结果与横向连接传递过来的且高层的第二特征图P″i+1进行单位加融合,最后通过对单位加融合结果进行3*3卷积处理,得到新第二细化新特征图N″i+1;
所述模型应用单元,用于将待处理MR图像导入所述MR脑肿瘤图像实例分割模型,输出实例分割结果。
本实施例第二方面提供的前述装置的工作过程、工作细节和技术效果,可以参见第一方面所述的MR脑肿瘤图像实例分割方法,于此不再赘述。
如图5所示,本实施例第三方面提供了一种执行第一方面所述MR脑肿瘤图像实例分割方法的计算机设备,包括有依次通信连接的存储器、处理器和收发器,其中,所述存储器用于存储计算机程序,所述收发器用于收发信息,所述处理器用于读取所述计算机程序,执行如第一方面所述的MR脑肿瘤图像实例分割方法。具体举例的,所述存储器可以但不限于包括随机存取存储器(Random-Access Memory,RAM)、只读存储器(Read-Only Memory,ROM)、闪存(Flash Memory)、先进先出存储器(First Input First Output,FIFO)和/或先进后出存储器(First Input Last Output,FILO)等等;所述处理器可以但不限于采用型号为STM32F105系列的微处理器。此外,所述计算机设备还可以但不限于包括有电源模块、显示屏和其它必要的部件。
本实施例第三方面提供的前述计算机设备的工作过程、工作细节和技术效果,可以参见第一方面所述的MR脑肿瘤图像实例分割方法,于此不再赘述。
本实施例第四方面提供了一种存储包含第一方面所述MR脑肿瘤图像实例分割方法的指令的计算机可读存储介质,即所述计算机可读存储介质上存储有指令,当所述指令在计算机上运行时,执行如第一方面所述的MR脑肿瘤图像实例分割方法。其中,所述计算机可读存储介质是指存储数据的载体,可以但不限于包括软盘、光盘、硬盘、闪存、优盘和/或记忆棒(Memory Stick)等计算机可读存储介质,所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。
本实施例第四方面提供的前述计算机可读存储介质的工作过程、工作细节和技术效果,可以参见第一方面所述的MR脑肿瘤图像实例分割方法,于此不再赘述。
本实施例第五方面提供了一种包含指令的计算机程序产品,当所述指令在计算机上运行时,使所述计算机执行如第一方面所述的MR脑肿瘤图像实例分割方法。其中,所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。
最后应说明的是,本发明不局限于上述可选的实施方式,任何人在本发明的启示下都可得出其他各种形式的产品。上述具体实施方式不应理解成对本发明的保护范围的限制,本发明的保护范围应当以权利要求书中界定的为准,并且说明书可以用于解释权利要求书。
Claims (10)
1.一种MR脑肿瘤图像实例分割方法,其特征在于,包括:
获取脑肿瘤公开数据训练集,其中,所述脑肿瘤公开数据训练集包含有多个MR脑肿瘤图像样本;
针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像;
将所有的所述样本图像送入集目标检测和实例分割于一体的Mask RCNN网络模型进行训练,得到训练完毕的MR脑肿瘤图像实例分割模型,其中,所述Mask RCNN网络模型由骨干backbone网络、颈部neck网络和头部head网络构成,所述backbone网络采用残差网络,所述neck网络采用特征金字塔网络FPN,所述head网络包括有预测框分类网络、边框bbox回归网络和mask分支网络;
所述残差网络用于在输入样本图像后,输出与该样本图像对应的六个第一特征图;
所述特征金字塔网络FPN的自顶而下通路Top-down pathway设计为:在从第三预测层P3至第七预测层P7的各个预测层中分别引入第一卷积注意力模块,以便采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,其中,i为自然数且有i∈[3,7];
所述残差网络还用于经过不同的降采样次数,使得所述第一细化新特征图Fi″在经过1*1降维处理和2倍上采样处理后,与由下一预测层输出的且经过1*1降维处理的另一第一细化新特征图Fi′-′1进行单位加融合,然后通过对单位加融合结果进行3*3卷积处理,得到第二特征图Pi′-′1;
在所述特征金字塔网络FPN中增加自底而上路径增强Bottom-up Path Augmentation,并设计为:先在所述各个预测层中分别引入第二卷积注意力模块,以便也采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图N″i,再自底而上地进行如下迭代处理得到用于预测框分类、边框bbox回归和mask分支生成的新第二细化新特征图:对低层的所述第二细化新特征图N″i进行一个步长为2的3*3卷积处理和下采样处理,然后将处理结果与横向连接传递过来的且高层的第二特征图P″i+1进行单位加融合,最后通过对单位加融合结果进行3*3卷积处理,得到新第二细化新特征图N″i+1;
将待处理MR图像导入所述MR脑肿瘤图像实例分割模型,输出实例分割结果。
2.如权利要求1所述的MR脑肿瘤图像实例分割方法,其特征在于,所述脑肿瘤公开数据训练集采用脑肿瘤公开数据集BRATS2018中的训练子集,其中,所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本分别有如下4种模态:T1模态、T2模态、T1ec模态和Flair模态。
3.如权利要求1所述的MR脑肿瘤图像实例分割方法,其特征在于,针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像,包括:将MR脑肿瘤图像样本标准化,使其具有零均值与单位标准差。
4.如权利要求3所述的MR脑肿瘤图像实例分割方法,其特征在于,针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像,还包括:
对标准化后的MR脑肿瘤图像样本进行如下方式的数据增扩处理:在各个通道上进行随机强度偏移处理,以及在三轴数据上进行随机旋转处理、镜像翻转处理、弹性形变处理和/或缩放处理。
5.如权利要求1所述的MR脑肿瘤图像实例分割方法,其特征在于,所述残差网络采用包含有七组卷积层的ResNet101网络结构;
在所述ResNet101网络结构中,第一组卷积层的大小是64*7*7且后接最大值池化层;
在所述ResNet101网络结构中,第二组卷积层包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为64*1*1、64*3*3和256*1*1;
在所述ResNet101网络结构中,第三组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为128*1*1、128*3*3和512*1*1;
在所述ResNet101网络结构中,第四组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为256*1*1、256*3*3和1024*1*1;
在所述ResNet101网络结构中,第五组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为512*1*1、512*3*3和2048*1*1;
在所述ResNet101网络结构中,第六组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为1024*1*1、1024*3*3和4096*1*1;
在所述ResNet101网络结构中,第七组卷积层也包含有三个步长均为1的卷积,并且这三个卷积的卷积核大小分别为2048*1*1、2048*3*3和8192*1*1。
6.如权利要求1所述的MR脑肿瘤图像实例分割方法,其特征在于,在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,包括:
式中,σ()表示归一化函数sigmoid()且用于根据神经网络的传播回馈不断更新权重系数以便指导模型对来自低层的肿瘤特征进行选择,AvgPool()表示平均池化函数,MaxPool()表示最大池化函数,MLP()表示多层感知器,表示对所述第一特征图Fi进行平均池化后得到的特征图,表示对所述第一特征图Fi进行最大池化后得到的特征图,W0和W1表示所述多层感知器MLP()的权重且有W0∈RC/r×C和W1∈RC×C/r,R表示实数,C/r表示所述多层感知器MLP()的第一层神经元个数,r表示预设减少率,C表示所述多层感知器MLP()的第二层神经元个数,所述多层感知器MLP()的激活函数采用线性整流函数Relu;
7.如权利要求1所述的MR脑肿瘤图像实例分割方法,其特征在于,在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图N″i,包括:
8.一种MR脑肿瘤图像实例分割装置,其特征在于,包括有依次通信连接的训练集获取单元、预处理单元、模型训练单元和模型应用单元;
所述训练集获取单元,用于获取脑肿瘤公开数据训练集,其中,所述脑肿瘤公开数据训练集包含有多个MR脑肿瘤图像样本;
所述预处理单元,用于针对所述多个MR脑肿瘤图像样本中的各个MR脑肿瘤图像样本,分别进行灰度数据归一化处理,得到对应的样本图像;
所述模型训练单元,用于将所有的所述样本图像送入集目标检测和实例分割于一体的Mask RCNN网络模型进行训练,得到训练完毕的MR脑肿瘤图像实例分割模型,其中,所述Mask RCNN网络模型由骨干backbone网络、颈部neck网络和头部head网络构成,所述backbone网络采用残差网络,所述neck网络采用特征金字塔网络FPN,所述head网络包括有预测框分类网络、边框bbox回归网络和mask分支网络;
所述残差网络用于在输入样本图像后,输出与该样本图像对应的六个第一特征图;
所述特征金字塔网络FPN的自顶而下通路Top-down pathway设计为:在从第三预测层P3至第七预测层P7的各个预测层中分别引入第一卷积注意力模块,以便采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入所述六个第一特征图中的第i个第一特征图Fi后,得到与该第i个第一特征图Fi对应的第一细化新特征图Fi″,其中,i为自然数且有i∈[3,7];
所述残差网络还用于经过不同的降采样次数,使得所述第一细化新特征图Fi″在经过1*1降维处理和2倍上采样处理后,与由下一预测层输出的且经过1*1降维处理的另一第一细化新特征图F″i-1进行单位加融合,然后通过对单位加融合结果进行3*3卷积处理,得到第二特征图P″i-1;
在所述特征金字塔网络FPN中增加自底而上路径增强Bottom-up Path Augmentation,并设计为:先在所述各个预测层中分别引入第二卷积注意力模块,以便也采用先通道后空间的策略,在空间和通道两个维度上进行重要特征加权,使得在输入第二特征图P″i-1后,得到与该第二特征图P″i-1对应的第二细化新特征图N″i,再自底而上地进行如下迭代处理得到用于预测框分类、边框bbox回归和mask分支生成的新第二细化新特征图:对低层的所述第二细化新特征图N″i进行一个步长为2的3*3卷积处理和下采样处理,然后将处理结果与横向连接传递过来的且高层的第二特征图P″i+1进行单位加融合,最后通过对单位加融合结果进行3*3卷积处理,得到新第二细化新特征图N″i+1;
所述模型应用单元,用于将待处理MR图像导入所述MR脑肿瘤图像实例分割模型,输出实例分割结果。
9.一种计算机设备,其特征在于,包括有依次通信连接的存储器、处理器和收发器,其中,所述存储器用于存储计算机程序,所述收发器用于收发信息,所述处理器用于读取所述计算机程序,执行如权利要求1~7中任意一项所述的MR脑肿瘤图像实例分割方法。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有指令,当所述指令在计算机上运行时,执行如权利要求1~7中任意一项所述的MR脑肿瘤图像实例分割方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111671947.1A CN114332463A (zh) | 2021-12-31 | 2021-12-31 | 一种mr脑肿瘤图像实例分割方法、装置、设备及存储介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111671947.1A CN114332463A (zh) | 2021-12-31 | 2021-12-31 | 一种mr脑肿瘤图像实例分割方法、装置、设备及存储介质 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN114332463A true CN114332463A (zh) | 2022-04-12 |
Family
ID=81020891
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202111671947.1A Pending CN114332463A (zh) | 2021-12-31 | 2021-12-31 | 一种mr脑肿瘤图像实例分割方法、装置、设备及存储介质 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114332463A (zh) |
Cited By (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN117036967A (zh) * | 2023-10-08 | 2023-11-10 | 江西师范大学 | 一种非视觉感知区域通道注意力的遥感图像描述方法 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111161275A (zh) * | 2018-11-08 | 2020-05-15 | 腾讯科技(深圳)有限公司 | 医学图像中目标对象的分割方法、装置和电子设备 |
CN111723845A (zh) * | 2020-05-19 | 2020-09-29 | 浙江工业大学 | 一种基于Mask轮廓的细胞图像分割方法 |
CN112785609A (zh) * | 2021-02-07 | 2021-05-11 | 重庆邮电大学 | 一种基于深度学习的cbct牙齿分割方法 |
CN113378813A (zh) * | 2021-05-28 | 2021-09-10 | 陕西大智慧医疗科技股份有限公司 | 基于注意力平衡特征金字塔的建模、目标检测方法及装置 |
CN113379773A (zh) * | 2021-05-28 | 2021-09-10 | 陕西大智慧医疗科技股份有限公司 | 基于双重注意力机制的分割模型建立、分割方法及装置 |
-
2021
- 2021-12-31 CN CN202111671947.1A patent/CN114332463A/zh active Pending
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111161275A (zh) * | 2018-11-08 | 2020-05-15 | 腾讯科技(深圳)有限公司 | 医学图像中目标对象的分割方法、装置和电子设备 |
CN111723845A (zh) * | 2020-05-19 | 2020-09-29 | 浙江工业大学 | 一种基于Mask轮廓的细胞图像分割方法 |
CN112785609A (zh) * | 2021-02-07 | 2021-05-11 | 重庆邮电大学 | 一种基于深度学习的cbct牙齿分割方法 |
CN113378813A (zh) * | 2021-05-28 | 2021-09-10 | 陕西大智慧医疗科技股份有限公司 | 基于注意力平衡特征金字塔的建模、目标检测方法及装置 |
CN113379773A (zh) * | 2021-05-28 | 2021-09-10 | 陕西大智慧医疗科技股份有限公司 | 基于双重注意力机制的分割模型建立、分割方法及装置 |
Non-Patent Citations (4)
Title |
---|
SANGHYUN WOO: "CBAM:Convolutional Block Attention Module", 《COMPUTER VISION - ECCV 2018》 * |
SHU LIU: "Path Aggregation Network for Instance Segmentation", 《2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION》 * |
XUAN NIE: "Attention Mask R-CNN for Ship Detection and Segmentation From Remote Sensing Images", 《IEEE ACCESS》 * |
郭启帆: "基于特征金字塔的多尺度特征融合网络", 《工程数学学报》 * |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN117036967A (zh) * | 2023-10-08 | 2023-11-10 | 江西师范大学 | 一种非视觉感知区域通道注意力的遥感图像描述方法 |
CN117036967B (zh) * | 2023-10-08 | 2024-01-19 | 江西师范大学 | 一种非视觉感知区域通道注意力的遥感图像描述方法 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
Cai et al. | A review of the application of deep learning in medical image classification and segmentation | |
Khouloud et al. | W-net and inception residual network for skin lesion segmentation and classification | |
JP7143008B2 (ja) | 深層学習に基づく医用画像検出方法及び装置、電子機器及びコンピュータプログラム | |
Yap et al. | Deep learning in diabetic foot ulcers detection: A comprehensive evaluation | |
Wu et al. | Discrimination and conversion prediction of mild cognitive impairment using convolutional neural networks | |
Moujahid et al. | Convolutional neural networks for multimodal brain MRI images segmentation: A comparative study | |
Wang et al. | Explainable diagnosis of secondary pulmonary tuberculosis by graph rank-based average pooling neural network | |
CN109949304B (zh) | 图像检测学习网络的训练和获取方法、图像检测装置和介质 | |
Qian et al. | Skin lesion classification using CNNs with grouping of multi-scale attention and class-specific loss weighting | |
CN115375711A (zh) | 基于多尺度融合的全局上下文关注网络的图像分割方法 | |
Huang et al. | DS-UNeXt: depthwise separable convolution network with large convolutional kernel for medical image segmentation | |
Jiang et al. | MFI-Net: A multi-resolution fusion input network for retinal vessel segmentation | |
Song et al. | Dual-branch network via pseudo-label training for thyroid nodule detection in ultrasound image | |
Nie et al. | Recent advances in diagnosis of skin lesions using dermoscopic images based on deep learning | |
Abbas et al. | Enhanced Skin Disease Diagnosis through Convolutional Neural Networks and Data Augmentation Techniques | |
CN114549538A (zh) | 一种基于空间信息与特征通道的脑肿瘤医学图像分割方法 | |
CN114663445A (zh) | 一种基于多尺度边缘感知的三维心脏图像分割方法 | |
Wang et al. | RFPNet: Reorganizing feature pyramid networks for medical image segmentation | |
Wu et al. | Continuous refinement-based digital pathology image assistance scheme in medical decision-making systems | |
CN114332463A (zh) | 一种mr脑肿瘤图像实例分割方法、装置、设备及存储介质 | |
Zhuang et al. | Class attention to regions of lesion for imbalanced medical image recognition | |
Zhou et al. | MCFA-UNet: multiscale cascaded feature attention U-Net for liver segmentation | |
Saber et al. | Multi-center, multi-vendor, and multi-disease cardiac image segmentation using scale-independent multi-gate UNET | |
Shao et al. | FCG-Net: an innovative full-scale connected network for thyroid nodule segmentation in ultrasound images | |
Rajput et al. | A triplanar ensemble model for brain tumor segmentation with volumetric multiparametric magnetic resonance images |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
RJ01 | Rejection of invention patent application after publication | ||
RJ01 | Rejection of invention patent application after publication |
Application publication date: 20220412 |