多模态医学图像的分割方法及终端设备与流程
本申请属于图像处理技术领域，尤其涉及一种多模态医学图像的分割方法及终端设备。
背景技术：
多模态医学图像分割，是从多种模态的医学图像(如多模态的计算机断层扫描图像和多模态磁共振图像等)中识别出器官或病灶位置的图像处理技术，通过该技术能够得到病灶的形状、病灶的体积等信息，对疾病的诊断、治疗和研究具有重要意义。
目前，多模态医学图像分割的主要方法是利用深度卷积神经网络进行图像分割。在该方法中，利用大量的训练数据对深度卷积神经网络进行训练，再利用训练后的深度卷积神经网络进行图像分割。在训练过程中，通常需要对每个训练数据进行人工标记，耗时耗力，降低了训练效率，进而影响了图像分割的效率；而且，人工标记的结果是由人为经验决定的，准确率较低，进而影响了图像分割结果的准确率。
技术实现要素：
本申请实施例提供了一种多模态医学图像的分割方法及终端设备，可以解决现有的多模态医学图像分割方法的效率较低、分割结果的准确率较低的问题。
第一方面，本申请实施例提供了一种多模态医学图像的分割方法，包括：
获取待分割的数据集，所述数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像；
利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果；
其中，所述训练后的分割模型为基于训练集进行半监督迭代训练后得到的分割模型，所述训练集包括至少一组已标记的数据集和多组未标记的数据集，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
在第一方面的一种可能的实现方式中，所述利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果，包括：
按照相同的预设划分规则，将所述待分割的数据集中各种模态的医学图像分别划分为m个区域块；
获取所述目标部位中的k个待分割子部位；
对于每个区域块，将各种模态的医学图像中所述区域块对应的图像块组合为一个图像矩阵，并利用当前的分割模型分别对所述图像矩阵进行各个待分割子部位的图像分割，得到所述区域块对应的实际分割结果；
将所述待分割的数据集中所有区域块对应的实际分割结果记为所述待分割的数据集对应的分割结果；
其中，每个区域块对应的实际分割结果包括k个概率矩阵，每个概率矩阵的大小与所述区域块的大小相同，第k个概率矩阵中的各元素表示所述区域块中各个体素属于第k个待分割子部位的概率值，1≤k≤k，k为大于1的正整数。
在第一方面的一种可能的实现方式中，在得到所述待分割的数据集对应的分割结果之后，所述方法还包括：
对所述待分割的数据集对应的分割结果中的每个概率矩阵分别进行二值化处理；
将所述待分割的数据集中同一个区域块对应的二值化处理后的概率矩阵组合成一个二值图像，得到m个二值图像，每个二值图像的大小和与所述二值图像对应的区域块的大小相同；
将所述m个二值图像按照所述预设划分规则进行组合，得到所述待分割的数据集的分割图像。
在第一方面的一种可能的实现方式中，在利用训练后的分割模型对所述待分割的数据集进行图像分割之前，所述方法还包括：
获取由至少一组已标记的数据集和多组未标记的数据集组成的训练集，其中，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像；
利用所述训练集对预设的分割模型进行半监督迭代训练，得到训练后的分割模型。
在第一方面的一种可能的实现方式中，所述利用所述训练集对预设的分割模型进行半监督迭代训练，得到训练后的分割模型，包括：
将所述训练集中所有未标记的数据集划分为n个子集合，每个子集合中包含至少两组未标记的数据集；
当进行第一次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对所述预设的分割模型进行训练，得到第一次训练后的分割模型，并利用所述第一次训练后的分割模型对第一个子集合中的各组数据集进行标记；
当进行第i次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对第i-1次训练后的分割模型进行训练，得到第i次训练后的分割模型，并利用所述第i次训练后的分割模型对第i个子集合中的各组数据集进行标记，其中，2≤i≤n；
当进行第n+1次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对第n次训练后的分割模型进行训练，得到训练后的分割模型。
在第一方面的一种可能的实现方式中，在任意一次半监督迭代训练的步骤中，利用当前的训练集中的一组已标记的数据集对当前的分割模型进行训练，包括：
按照相同的预设划分规则，将所述已标记的数据集中各种模态的医学图像分别划分为m个区域块；
获取所述目标部位中的l个待分割子部位；
对于每个区域块，将各种模态的医学图像中所述区域块对应的图像块组合为一个图像矩阵，并利用当前的分割模型分别对所述图像矩阵进行各个待分割子部位的图像分割，得到所述区域块对应的样本分割结果；
利用所述区域块对应的样本分割结果计算当前的分割模型的损失值，并利用所述损失值对当前的分割模型进行参数更新；
其中，所述区域块对应的样本分割结果包括l个概率矩阵，每个概率矩阵的大小与所述区域块的大小相同，第l个概率矩阵中的各元素表示所述区域块中各个体素属于第l个待分割子部位的概率值，1≤l≤l，l为大于1的正整数。
在第一方面的一种可能的实现方式中，所述利用所述区域块对应的样本分割结果计算当前的分割模型的损失值，包括：
利用所述区域块对应的标记，分别计算所述区域块对应的样本分割结果中每个概率矩阵的损失系数；
将各个概率矩阵的损失系数进行加权求和，得到当前的分割模型的损失值。
在第一方面的一种可能的实现方式中，所述分割模型包括依次顺序连接的第一卷积层、第一子模型、第二子模型、第一反卷积层和第二卷积层；
其中，所述第一子模型包括4个依次顺序连接的第一子块，每个第一子块中包括依次顺序连接的一个池化层、一个残差卷积模块和一个注意力机制模块；
所述第二子模型包括3个依次顺序连接的第二子块，每个第二子块中包括依次顺序连接的一个反卷积层和一个残差卷积模块；
第j个第一子块的输出端还连接第4-j个第二子块中反卷积层的输出端，1≤j≤3。
第二方面，本申请实施例提供了一种多模态医学图像的分割装置，包括：
获取单元，用于获取待分割的数据集，所述数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像；
分割单元，用于利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果；
其中，所述训练后的分割模型为进行半监督迭代训练后得到的分割模型。
第三方面，本申请实施例提供了一种终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如上述第一方面中任一项所述的多模态医学图像的分割方法。
第四方面，本申请实施例提供了一种计算机可读存储介质，本申请实施例提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如上述第一方面中任一项所述的多模态医学图像的分割方法。
第五方面，本申请实施例提供了一种计算机程序产品，当计算机程序产品在终端设备上运行时，使得终端设备执行上述第一方面中任一项所述的多模态医学图像的分割方法。
可以理解的是，上述第二方面至第五方面的有益效果可以参见上述第一方面中的相关描述，在此不再赘述。
本申请实施例与现有技术相比存在的有益效果是：
本申请实施例通过获取待分割的数据集，所述数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像，多模态的医学图像中包含有更完整、更全面的病灶信息，为后续图像分割提供可靠依据；利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果，通过上述方法，将待分割的数据集直接输入训练后的分割模型即可得到分割结果，能够有效提高图像分割的效率；其中，所述训练后的分割模型为基于训练集进行半监督迭代训练后得到的分割模型，所述训练集包括至少一组已标记的数据集和多组未标记的数据集，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像；在半监督迭代训练的过程中，只需少量的标记数据即可完成分割模型的训练，无需进行大量的人工标记，省时省力，同时避免了人为经验对标记结果的影响，进而保证了分割模型的精确度。通过上述方法，在提高多模态医学图像分割效率的同时，有效提高了多模态医学图像分割结果的准确率。
附图说明
为了更清楚地说明本申请实施例中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动性的前提下，还可以根据这些附图获得其他的附图。
图1是本申请一实施例提供的医学图像分割系统的示意图；
图2是本申请一实施例提供的多模态医学图像的分割方法的流程示意图；
图3是本申请一实施例提供的4种模态的mri切片图像的示意图；
图4是本申请一实施例提供的图像分割方法的示意图；
图5是本申请一实施例提供的分割模型的训练方法的流程示意图；
图6是本申请一实施例提供的分割模型的示意图；
图7是本申请一实施例提供的多模态医学图像的分割装置的结构框图；
图8是本申请一实施例提供的终端设备的结构示意图。
具体实施方式
以下描述中，为了说明而不是为了限定，提出了诸如特定系统结构、技术之类的具体细节，以便透彻理解本申请实施例。然而，本领域的技术人员应当清楚，在没有这些具体细节的其它实施例中也可以实现本申请。在其它情况中，省略对众所周知的系统、装置、电路以及方法的详细说明，以免不必要的细节妨碍本申请的描述。
应当理解，当在本申请说明书和所附权利要求书中使用时，术语“包括”指示所描述特征、整体、步骤、操作、元素和/或组件的存在，但并不排除一个或多个其它特征、整体、步骤、操作、元素、组件和/或其集合的存在或添加。
另外，在本申请说明书和所附权利要求书的描述中，术语“第一个”、“第二个”、“第三个”等仅用于区分描述，而不能理解为指示或暗示相对重要性。
在本申请说明书中描述的参考“一个实施例”或“一些实施例”等意味着在本申请的一个或多个实施例中包括结合该实施例描述的特定特征、结构或特点。由此，在本说明书中的不同之处出现的语句“在一个实施例中”、“在一些实施例中”、“在其他一些实施例中”、“在另外一些实施例中”等不是必然都参考相同的实施例，而是意味着“一个或多个但不是所有的实施例”，除非是以其他方式另外特别强调。
先介绍本申请实施例的一个应用场景。参见图1，为本申请实施例提供的医学图像分割系统的示意图。如图1所示，医学图像分割系统包括医学图像采集设备101和终端设备102。其中，医学图像采集设备可以是计算机断层扫描(computedtomography，ct)仪、核磁共振成像(magneticresonanceimaging，mri)仪等等。终端设备可以是手机、电脑、云处理器等。医学图像采集设备负责采集多模态医学图像，并将采集到的多模态医学图像发送给终端设备；终端设备利用本申请实施例中的多模态医学图像的分割方法对接收到的多模态医学图像进行图像分割。
终端设备可以通过有线或无线的方式与医学图像采集设备通信连接。当二者无线连接时，可以实现远程的图像分割，增加了系统的适应性。
图2示出了本申请一实施例提供的多模态医学图像的分割方法的流程示意图，作为示例而非限定，分割方法可以包括以下步骤：
s201，获取待分割的数据集，所述数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
示例性的，拍摄对象可以指病人，目标部位可以是待诊断的器官或人体部位。例如，待分割的数据集中包括一个病人脑部的多模态的医学图像。换句话说，一组数据集中包含多张医学图像，这些医学图像都是对同一个病人的同一个目标部位拍摄得到的，每张医学图像对应一种模态。
在实际应用中，医学图像可以是mri图像。mri图像是由多张切片图像组成的三维图像，其中，切片图像为二维图像。多模态mri图像为在mri成像过程中利用不同的脉冲序列生成的一系列mri图像，即每种模态对应一张mri图像。参见图3，为本申请实施例提供的4种模态的mri切片图像的示意图。如图3所示，4种模态分别为flair、t1、t1c和t2。图3为4种模态的mri图像中相同层对应的切片图像(即二维图像)，图中灰度值越高的部分(较白的部分)表示病灶区。可以看出，不同模态的图像中病灶区域的亮暗程度不一样。从不同模态的mri图像中截取相同层的切片图像进行对比，就可以观察出病灶的形态、体积等信息。
多模态mri图像能够更清楚、更完整地反映拍摄对象的目标部位的病灶信息。
s202，利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果。
其中，所述训练后的分割模型为基于训练集进行半监督迭代训练后得到的分割模型，所述训练集包括至少一组已标记的数据集和多组未标记的数据集，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
半监督迭代训练属于半监督学习，即使用少量已标记数据和大量未标记数据进行训练。现有技术中，在对分割模型进行训练时，需要对训练数据进行全部标记，费时费力。本申请实施例中基于半监督学习，只需利用少量的已标记数据即可实现对分割模型的精确，省时省力。
在一个实施例中，s202可以包括以下步骤：
s21，按照相同的预设划分规则，将所述待分割的数据集中各种模态的医学图像分别划分为m个区域块。
在实际应用中，分割模型的数据处理能力是有限的，而医学图像通常包含较多的体素信息，即医学图像可能超过了分割模型的数据处理能力。这时需要将医学图像划分为低于分割模型的数据处理能力的小图像块。
预设划分规则可以是根据实际需要预先设定的。设定了划分规则后，每一张医学图像都基于这个划分规则进行划分。
示例性的，参见图4，为本申请实施例提供的图像分割方法的示意图。如图4所示，一组数据集中包含4种模态的医学图像(每个立方体表示一张模态图，即一种模态对应的医学图像，每张模态图的大小为256×128×256)。预设划分规则为，分别以坐标点(0,0,0)、(128,0,0)、(0,0,128)和(128,0,128)为起点，延x轴正方向、y轴正方向和z轴正方向截取尺寸为128×128×128的区域块。如图4中步骤(a)所示，基于上述规则，每张模态图均被划分为4个区域块1、2、3、4，且每个区域块的大小为128×128×128。
s22，获取所述目标部位中的k个待分割子部位。
示例性的，假设目标部位为人脑，待分割子部位可以是脑叶、小脑、丘脑等等。待分割子部位可以根据实际需要获取。例如，某个病人的病灶位于脑叶和丘脑部分，那么可以只将这两个部分作为待分割子部位。
s23，对于每个区域块，将各种模态的医学图像中所述区域块对应的图像块组合为一个图像矩阵，并利用当前的分割模型分别对所述图像矩阵进行各个待分割子部位的图像分割，得到所述区域块对应的实际分割结果。
可以延z轴方向将相应的图像块摞在一起。如图4中步骤(b)所示，将4张模态图中区域块1对应的图像块组合成图像矩阵1，将4张模态图中区域块2对应的图像块组合成图像矩阵2，将4张模态图中区域块3对应的图像块组合成图像矩阵3，将4张模态图中区域块4对应的图像块组合成图像矩阵4。一共得到4个图像矩阵，每个图像矩阵的大小为128×128×(128×4)。当然，在实际应用中，也可以由其他的组合方式，上述只是示例，并不用于具体限定。
其中，每个区域块对应的实际分割结果包括k个概率矩阵，每个概率矩阵的大小与所述区域块的大小相同，第k个概率矩阵中的各元素表示所述区域块中各个体素属于第k个待分割子部位的概率值，1≤k≤k，k为大于1的正整数。
示例性的，第k个概率矩阵中(0,0,0)位置上的元素表示区域块(0,0,0)位置上的体素属于第k个待分割子部位的概率值。
参见图4中的步骤(c)，分别将各个图像矩阵输入到分割模型中。对于每个图像矩阵，对应的可以得到分割模型输出的3个概率矩阵(此时待分割子部位的数量为3)，每个概率矩阵的大小均为128×128×128。
s24，将所述待分割的数据集中所有区域块对应的实际分割结果记为所述待分割的数据集对应的分割结果。
共有m个区域块，每个区域块对应k个概率矩阵，因此，一个数据集对应的分割结果中包括m×k个概率矩阵。
此时，得到的分割结果仅是概率矩阵，为了得到直观的分割图像，在一个实施例中，在步骤s202得到待分割的数据集对应的分割结果之后，多模态医学图像的分割方法还可以包括以下步骤：
s203，对所述待分割的数据集对应的分割结果中的每个概率矩阵分别进行二值化处理。
实际应用中，可以预先设定阈值，将概率矩阵中的每个元素分别和阈值进行比较。当元素大于或等于阈值时，将该元素设置为第一预设值；当元素小于阈值时，将该元素设置为第二预设值。示例性的，第一预设值为1，第二预设值为0，这样得到的二值图像就可以清楚地展示出待分割子部位的边界以及所占的区域。参见图4中的步骤(d)。
s204，将所述待分割的数据集中同一个区域块对应的二值化处理后的概率矩阵组合成一个二值图像，得到m个二值图像，每个二值图像的大小和与所述二值图像对应的区域块的大小相同。
参见图4中的步骤(e)，一个区域块对应3个概率矩阵(即一个图像矩阵对应3个概率矩阵)，将这3个概率矩阵分别进行二值化处理后组合成一个二值图像，图4中二值图像的大小为128×128×128，与区域块大小相同。如图4所示，区域块1对应二值图像1，区域块2对应二值图像2，区域块3对应二值图像3，区域块4对应二值图像4。
到此步骤，实际得到了每个区域块对应的分割图像，并不是完整的分割图像。
s205，将所述m个二值图像按照所述预设划分规则进行组合，得到所述待分割的数据集的分割图像。
将各个区域块对应的分割图像按照之前划分时的规则组合在一起，就可以得到完整的分割图像。这时的完整的分割图像的大小与分割前的各张医学图像的大小相同。如图4步骤(f)所示，最后得到的分割图像的大小为256×128×256。且分割图像中，区域块1的位置为二值图像1，区域块2的位置为二值图像2，区域块3的位置为二值图像3，区域块4的位置为二值图像4。
本申请实施例通过获取待分割的数据集，所述数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像，多模态的医学图像中包含有更完整、更全面的病灶信息，为后续图像分割提供可靠依据；利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果，通过上述方法，将待分割的数据集直接输入训练后的分割模型即可得到分割结果，能够有效提高图像分割的效率；其中，所述训练后的分割模型为基于训练集进行半监督迭代训练后得到的分割模型，所述训练集包括至少一组已标记的数据集和多组未标记的数据集，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像；在半监督迭代训练的过程中，只需少量的标记数据即可完成分割模型的训练，无需进行大量的人工标记，省时省力，同时避免了人为经验对标记结果的影响，进而保证了分割模型的精确度。通过上述方法，在提高多模态医学图像分割效率的同时，有效提高了多模态医学图像分割结果的准确率。
图5示出了本申请一实施例提供的分割模型的训练方法的流程示意图，作为示例而非限定，分割模型的训练方法可以包括以下步骤：
s501，获取由至少一组已标记的数据集和多组未标记的数据集组成的训练集，其中，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
已标记的数据集中的每张医学图像上都包含有待分割子部位的标记，可以只包含一个待分割子部位的标记，也可以包括多个待分割子部位的标记。
哪个待分割子部位带有标记，训练后的分割模型就可以分割出该待分割子部位。如果训练集中不包含已标记出待分割子部位a的数据集，那么训练后的分割模型也不能分割出待分割子部位a。
s502，利用所述训练集对预设的分割模型进行半监督迭代训练，得到训练后的分割模型。
在一个实施例中，可以分批地进行迭代训练，这样可以增加迭代训练的次数，进而可以提高训练后的分割模型的精度。具体方法如下。
s51，将所述训练集中所有未标记的数据集划分为n个子集合，每个子集合中包含至少两组未标记的数据集。
在实际应用中，每个子集合中包含的未标记的数据集的组数与已标记的数据集的组数相等，或两者差值在预设范围内。这样可以避免训练过程中的数据冗余或数据不足。
s52，当进行第一次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对所述预设的分割模型进行训练，得到第一次训练后的分割模型，并利用所述第一次训练后的分割模型对第一个子集合中的各组数据集进行标记。
s53，当进行第i次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对第i-1次训练后的分割模型进行训练，得到第i次训练后的分割模型，并利用所述第i次训练后的分割模型对第i个子集合中的各组数据集进行标记，其中，2≤i≤n。
s54，当进行第n+1次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对第n次训练后的分割模型进行训练，得到训练后的分割模型。
示例性的，已标记的数据集为dl，将未标记的数据集划分为3个，分别为d1、d2和d3。进行第一次半监督迭代训练时，利用dl对预设的分割模型进行训练得到第一次训练后的分割模型m1，利用m1对d1中的各组数据集进行标记；进行第二次半监督迭代训练时，利用dl和d1对m1进行训练(当前的训练集中已标记的数据集包括dl和d1)得到第二次训练后的分割模型m2，利用m2对d2中的各组数据集进行标记；进行第三次半监督迭代训练时，利用dl、d1和d2对m2进行训练(当前的训练集中已标记的数据集包括dl、d1和d2)得到第三次训练后的分割模型m3，利用m3对d3中的各组数据集进行标记；进行第四次半监督迭代训练时，利用dl、d1、d2和d3对m3进行训练，得到训练后的分割模型。
在一个实施例中，在任意一次半监督迭代训练的步骤中，利用当前的训练集中的一组已标记的数据集对当前的分割模型进行训练，包括：
s61，按照相同的预设划分规则，将所述已标记的数据集中各种模态的医学图像分别划分为m个区域块。
这里的预设划分规则，与图2分割方法步骤中的预设划分规则相同。即训练的时候用哪种划分规则，在实际分割图像的时候也用哪种划分规则。这样，可以保证分割的准确率。
划分方法与s21中的步骤相同，具体可参见s21中的描述，在此不再赘述。
s62，获取所述目标部位中的l个待分割子部位。
原则上，l应大于或等于k。例如，在训练过程中，针对3个待分割子部位a、b、c进行了训练，那么在分割过程中，只能针对a、b、c这3个待分割子部位中的至少一个进行分割。
s63，对于每个区域块，将各种模态的医学图像中所述区域块对应的图像块组合为一个图像矩阵，并利用当前的分割模型分别对所述图像矩阵进行各个待分割子部位的图像分割，得到所述区域块对应的样本分割结果。
其中，所述区域块对应的样本分割结果包括l个概率矩阵，每个概率矩阵的大小与所述区域块的大小相同，第l个概率矩阵中的各元素表示所述区域块中各个体素属于第l个待分割子部位的概率值，1≤l≤l，l为大于1的正整数。
s63中的步骤与s23中的步骤相同，具体可参见s23中的描述，在此不再赘述。
s64，利用所述区域块对应的样本分割结果计算当前的分割模型的损失值，并利用所述损失值对当前的分割模型进行参数更新。
可选的，步骤s64中计算损失值可以包括：
s641，利用所述区域块对应的标记，分别计算所述区域块对应的样本分割结果中每个概率矩阵的损失系数。
可选的，可以利用公式
在实际应用中，也可以先对区域块对应的各个概率矩阵进行二值化处理，再利用二值化处理后的概率矩阵计算损失值。
s642，将各个概率矩阵的损失系数进行加权求和，得到当前的分割模型的损失值。
可选的，可以利用公式
通过上述计算损失值的方法，即采用类间加权值的方法，可以均衡不同待分割子部位对整体损失值的影响。
本申请实施例通过获取由至少一组已标记的数据集和多组未标记的数据集组成的训练集，其中，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像；利用所述训练集对预设的分割模型进行半监督迭代训练，得到训练后的分割模型。通过上述方法，只需利用较少的已标记数据集，即可实现对分割模型的训练，同时能够保证训练后的分割模型具有较高的分割准确率。
图6示出了本申请一实施例提供的分割模型的示意图，如图6所示，在本申请实施例中，可以采用如下结构的分割模型。
分割模型包括依次顺序连接的第一卷积层(如图6中最左侧的conv)、第一子模型、第二子模型、第一反卷积层(如图6中的最右侧的deconv)和第二卷积层(如图6中的最右侧的conv)。
其中，所述第一子模型包括4个依次顺序连接的第一子块，每个第一子块中包括依次顺序连接的一个池化层(pooling)、一个残差卷积模块(resblock)和一个注意力机制模块(scseblock)。
所述第二子模型包括3个依次顺序连接的第二子块，每个第二子块中包括依次顺序连接的一个反卷积层(deconv)和一个残差卷积模块(resblock)。
第j个第一子块的输出端还连接第4-j个第二子块中反卷积层的输出端，1≤j≤3。
在利用分割模型对图像进行图像分割时，将图像输入第一卷积层进行第一次卷积处理，第一卷积层将得到第一次卷积后的图像输入到第一子模型中。
在第一子模型中，进行4次第一子处理。其中，第一次子处理为，利用第一个第一子块中的池化层对第一次卷积后的图像进行下采样处理得到第一个下采样图像，利用第一个第一子块中的残差卷积模型对第一下采样图像进行残差卷积处理得到第一个残差图像，利用第一个第一子块中的注意力机制模块对第一残差图像进行加权处理得到第一个加权图像。第s次子处理为，利用第s个第一子块中的池化层对第s－1个加权图像进行下采样处理得到第s个下采样图像，利用第s个第一子块中的残差卷积模型对第s个下采样图像进行残差卷积处理得到第s个残差图像，利用第s个第一子块中的注意力机制模块对第s个残差图像进行加权处理得到第s个加权图像，s为大于1且小于5的正整数。
利用第一子模型得到的第4个加权图像作为第二子模型的输入。在第二子模型中，进行3次第二子处理。其中，第一次子处理为，利用第一个第二子块中的反卷积层对第4个加权图像进行反卷积处理得到第一个反卷积图像，利用第一个第二子块中的残差卷积模块对第一个反卷积图像与第3次第一子处理得到的第3个加权图像之和进行残差卷积处理得到第5个残差图像。第t次第二子处理为，利用第t个第二子块中的反卷积层对第t+3个残差图像进行反卷积处理得到第t个反卷积图像，利用第t个第二子块中的残差卷积模块对第t个反卷积图像与第4-t次第一子处理得到的第4-t个加权图像之和进行残差卷积处理得到第t+4个残差图像，t为大于1且小于4的正整数。
第二子模型得到的第7个残差图像作为第一反卷积层的输入，利用第一反卷积层对第7个残差图像进行反卷积处理得到第4个反卷积图像。
第4个反卷积图像作为第二卷积层的输入，利用第二卷积层对第4个反卷积图像进行卷积处理得到最后的分割结果。
分割模型中加入注意力机制模块，可以较好地避免训练时出现梯度消失或梯度爆炸的情况，并且该注意力机制模块在训练阶段可以自动学习加权特征，提高了分割模型的鲁棒性和自适应性。
应理解，上述实施例中各步骤的序号的大小并不意味着执行顺序的先后，各过程的执行顺序应以其功能和内在逻辑确定，而不应对本申请实施例的实施过程构成任何限定。
对应于上文实施例所述的方法，图7示出了本申请实施例提供的多模态医学图像的分割装置的结构框图，为了便于说明，仅示出了与本申请实施例相关的部分。
参照图7，该装置7包括：
获取单元71，用于获取待分割的数据集，所述数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
分割单元72，用于利用训练后的分割模型对所述待分割的数据集进行图像分割，得到所述待分割的数据集对应的分割结果。
其中，所述训练后的分割模型为基于训练集进行半监督迭代训练后得到的分割模型，所述训练集包括至少一组已标记的数据集和多组未标记的数据集，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
可选的，分割单元72包括：
划分模块，用于按照相同的预设划分规则，将所述待分割的数据集中各种模态的医学图像分别划分为m个区域块。
获取模块，用于获取所述目标部位中的k个待分割子部位。
分割模块，用于对于每个区域块，将各种模态的医学图像中所述区域块对应的图像块组合为一个图像矩阵，并利用当前的分割模型分别对所述图像矩阵进行各个待分割子部位的图像分割，得到所述区域块对应的实际分割结果。
结果模块，用于将所述待分割的数据集中所有区域块对应的实际分割结果记为所述待分割的数据集对应的分割结果。
其中，每个区域块对应的实际分割结果包括k个概率矩阵，每个概率矩阵的大小与所述区域块的大小相同，第k个概率矩阵中的各元素表示所述区域块中各个体素属于第k个待分割子部位的概率值，1≤k≤k，k为大于1的正整数。
可选的，装置7还包括：
二值化单元73，用于在得到所述待分割的数据集对应的分割结果之后，对所述待分割的数据集对应的分割结果中的每个概率矩阵分别进行二值化处理。
组合单元74，用于将所述待分割的数据集中同一个区域块对应的二值化处理后的概率矩阵组合成一个二值图像，得到m个二值图像，每个二值图像的大小和与所述二值图像对应的区域块的大小相同。
拼接单元75，用于将所述m个二值图像按照所述预设划分规则进行组合，得到所述待分割的数据集的分割图像。
可选的，装置7还包括：
数据获取单元76，用于获取由至少一组已标记的数据集和多组未标记的数据集组成的训练集，其中，每组数据集中包括对同一拍摄对象的目标部位拍摄得到的多种模态的医学图像。
训练单元77，用于利用所述训练集对预设的分割模型进行半监督迭代训练，得到训练后的分割模型。
可选的，训练单元77还用于：
将所述训练集中所有未标记的数据集划分为n个子集合，每个子集合中包含至少两组未标记的数据集。
当进行第一次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对所述预设的分割模型进行训练，得到第一次训练后的分割模型，并利用所述第一次训练后的分割模型对第一个子集合中的各组数据集进行标记。
当进行第i次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对第i-1次训练后的分割模型进行训练，得到第i次训练后的分割模型，并利用所述第i次训练后的分割模型对第i个子集合中的各组数据集进行标记，其中，2≤i≤n。
当进行第n+1次半监督迭代训练时，利用当前的训练集中所有已标记的数据集对第n次训练后的分割模型进行训练，得到训练后的分割模型。
可选的，训练单元77还用于：
按照相同的预设划分规则，将所述已标记的数据集中各种模态的医学图像分别划分为m个区域块。
获取所述目标部位中的l个待分割子部位。
对于每个区域块，将各种模态的医学图像中所述区域块对应的图像块组合为一个图像矩阵，并利用当前的分割模型分别对所述图像矩阵进行各个待分割子部位的图像分割，得到所述区域块对应的样本分割结果。
利用所述区域块对应的样本分割结果计算当前的分割模型的损失值，并利用所述损失值对当前的分割模型进行参数更新。
其中，所述区域块对应的样本分割结果包括l个概率矩阵，每个概率矩阵的大小与所述区域块的大小相同，第l个概率矩阵中的各元素表示所述区域块中各个体素属于第l个待分割子部位的概率值，1≤l≤l，l为大于1的正整数。
可选的，训练单元77还用于：
利用所述区域块对应的标记，分别计算所述区域块对应的样本分割结果中每个概率矩阵的损失系数。
将各个概率矩阵的损失系数进行加权求和，得到当前的分割模型的损失值。
可选的，所述分割模型包括依次顺序连接的第一卷积层、第一子模型、第二子模型、第一反卷积层和第二卷积层。
其中，所述第一子模型包括4个依次顺序连接的第一子块，每个第一子块中包括依次顺序连接的一个池化层、一个残差卷积模块和一个注意力机制模块。
所述第二子模型包括3个依次顺序连接的第二子块，每个第二子块中包括依次顺序连接的一个反卷积层和一个残差卷积模块。
第j个第一子块的输出端还连接第4-j个第二子块中反卷积层的输出端，1≤j≤3。
需要说明的是，上述装置/单元之间的信息交互、执行过程等内容，由于与本申请方法实施例基于同一构思，其具体功能及带来的技术效果，具体可参见方法实施例部分，此处不再赘述。
另外，图7所示的装置可以是内置于现有的终端设备内的软件单元、硬件单元、或软硬结合的单元，也可以作为独立的挂件集成到所述终端设备中，还可以作为独立的终端设备存在。
所属领域的技术人员可以清楚地了解到，为了描述的方便和简洁，仅以上述各功能单元、模块的划分进行举例说明，实际应用中，可以根据需要而将上述功能分配由不同的功能单元、模块完成，即将所述装置的内部结构划分成不同的功能单元或模块，以完成以上描述的全部或者部分功能。实施例中的各功能单元、模块可以集成在一个处理单元中，也可以是各个单元单独物理存在，也可以两个或两个以上单元集成在一个单元中，上述集成的单元既可以采用硬件的形式实现，也可以采用软件功能单元的形式实现。另外，各功能单元、模块的具体名称也只是为了便于相互区分，并不用于限制本申请的保护范围。上述系统中单元、模块的具体工作过程，可以参考前述方法实施例中的对应过程，在此不再赘述。
图8为本申请一实施例提供的终端设备的结构示意图。如图8所示，该实施例的终端设备8包括：至少一个处理器80(图8中仅示出一个)处理器、存储器81以及存储在所述存储器81中并可在所述至少一个处理器80上运行的计算机程序82，所述处理器80执行所述计算机程序82时实现上述任意各个多模态医学图像的分割方法实施例中的步骤。
所述终端设备可以是桌上型计算机、笔记本、掌上电脑及云端服务器等计算设备。该终端设备可包括，但不仅限于，处理器、存储器。本领域技术人员可以理解，图8仅仅是终端设备8的举例，并不构成对终端设备8的限定，可以包括比图示更多或更少的部件，或者组合某些部件，或者不同的部件，例如还可以包括输入输出设备、网络接入设备等。
所称处理器80可以是中央处理单元(centralprocessingunit，cpu)，该处理器80还可以是其他通用处理器、数字信号处理器(digitalsignalprocessor，dsp)、专用集成电路(applicationspecificintegratedcircuit，asic)、现成可编程门阵列(field-programmablegatearray，fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
所述存储器81在一些实施例中可以是所述终端设备8的内部存储单元，例如终端设备8的硬盘或内存。所述存储器81在另一些实施例中也可以是所述终端设备8的外部存储设备，例如所述终端设备8上配备的插接式硬盘，智能存储卡(smartmediacard,smc)，安全数字(securedigital,sd)卡，闪存卡(flashcard)等。进一步地，所述存储器81还可以既包括所述终端设备8的内部存储单元也包括外部存储设备。所述存储器81用于存储操作系统、应用程序、引导装载程序(bootloader)、数据以及其他程序等，例如所述计算机程序的程序代码等。所述存储器81还可以用于暂时地存储已经输出或者将要输出的数据。
本申请实施例还提供了一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现可实现上述各个方法实施例中的步骤。
本申请实施例提供了一种计算机程序产品，当计算机程序产品在移动终端上运行时，使得移动终端执行时实现可实现上述各个方法实施例中的步骤。
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本申请实现上述实施例方法中的全部或部分流程，可以通过计算机程序来指令相关的硬件来完成，所述的计算机程序可存储于一计算机可读存储介质中，该计算机程序在被处理器执行时，可实现上述各个方法实施例的步骤。其中，所述计算机程序包括计算机程序代码，所述计算机程序代码可以为源代码形式、对象代码形式、可执行文件或某些中间形式等。所述计算机可读介质至少可以包括：能够将计算机程序代码携带到装置/终端设备的任何实体或装置、记录介质、计算机存储器、只读存储器(rom，read-onlymemory)、随机存取存储器(ram，randomaccessmemory)、电载波信号、电信信号以及软件分发介质。例如u盘、移动硬盘、磁碟或者光盘等。在某些司法管辖区，根据立法和专利实践，计算机可读介质不可以是电载波信号和电信信号。
在上述实施例中，对各个实施例的描述都各有侧重，某个实施例中没有详述或记载的部分，可以参见其它实施例的相关描述。
本领域普通技术人员可以意识到，结合本文中所公开的实施例描述的各示例的单元及算法步骤，能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行，取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能，但是这种实现不应认为超出本申请的范围。
在本申请所提供的实施例中，应该理解到，所揭露的装置/网络设备和方法，可以通过其它的方式实现。例如，以上所描述的装置/网络设备实施例仅仅是示意性的，例如，所述模块或单元的划分，仅仅为一种逻辑功能划分，实际实现时可以有另外的划分方式，例如多个单元或组件可以结合或者可以集成到另一个系统，或一些特征可以忽略，或不执行。另一点，所显示或讨论的相互之间的耦合或直接耦合或通讯连接可以是通过一些接口，装置或单元的间接耦合或通讯连接，可以是电性，机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的，作为单元显示的部件可以是或者也可以不是物理单元，即可以位于一个地方，或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
以上所述实施例仅用以说明本申请的技术方案，而非对其限制；尽管参照前述实施例对本申请进行了详细的说明，本领域的普通技术人员应当理解：其依然可以对前述各实施例所记载的技术方案进行修改，或者对其中部分技术特征进行等同替换；而这些修改或者替换，并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围，均应包含在本申请的保护范围之内。
- 还没有人留言评论。精彩留言会获得点赞！