首页 / 专利库 / 电脑图像 / 像素 / 一种待标注图像确定的方法、模型训练的方法及装置

一种待标注图像确定的方法、模型训练的方法及装置

阅读:651发布:2024-02-06

专利汇可以提供一种待标注图像确定的方法、模型训练的方法及装置专利检索,专利查询,专利分析的服务。并且本 申请 公开了一种待标注图像确定的方法,该方法应用于 人工智能 领域,包括:获取原始图像以及自 编码器 集合,其中,原始图像为未经过标注的图像,自编码器集合包括N个自编码器;通过自编码器集合获取原始图像所对应的编码图像集合,编码图像集合包括N个编码图像,编码图像与自编码器之间具有对应关系;通过 图像分割 网络获取编码图像集合以及原始图像所对应的分割结果集合,图像分割网络包括M个子图像分割网络,分割结果集合包括[(N+1)*M]个分割结果;根据分割结果集合确定原始图像所对应的不确定度。本申请公开了一种模型训练的方法。本申请能够使得输出的结果变化更多,从而提升不确定度的计算合理性,进而有助于选择更加准确的待标注图像。,下面是一种待标注图像确定的方法、模型训练的方法及装置专利的具体信息内容。

1.一种待标注图像确定的方法,其特征在于,包括:
获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
根据所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
2.根据权利要求1所述的方法,其特征在于,所述通过自编码器集合获取所述原始图像所对应的编码图像集合,包括:
通过第一自编码器获取所述原始图像所对应的第一编码图像;
通过第二自编码器获取所述原始图像所对应的第二编码图像,其中,所述第二自编码器与所述第一自编码器属于两种不同类型的自编码器,所述第二编码图像和所述第一编码图像属于两个不同的图像。
3.根据权利要求1所述的方法,其特征在于,所述通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,包括:
通过第一子图像分割网络获取第一分割结果,其中,所述第一分割结果包括(N+1)个第一分割子结果,所述第一分割子结果与所述编码图像和所述原始图像具有对应关系;
通过第二子图像分割网络获取第二分割结果,其中,所述第二分割结果包括(N+1)个第二分割子结果,所述第二分割子结果与所述编码图像和所述原始图像具有对应关系,所述第二子图像分割网络与所述第一子图像分割网络属于两种不同类型的子图像分割网络。
4.根据权利要求1至3中任一项所述的方法,其特征在于,所述根据所述分割结果集合确定所述原始图像所对应的不确定度,包括:
获取所述原始图像中每个像素点所对应的灰度值;
根据所述分割结果集合以及所述原始图像中每个像素点所对应的灰度值,确定所述原始图像中每个像素点所对应的熵集合,其中,所述熵集合包括[(N+1)*M]个熵;
根据所述原始图像中每个像素点所对应的熵集合,计算得到所述原始图像所对应的不确定度。
5.根据权利要求1至3中任一项所述的方法,其特征在于,所述根据所述分割结果集合确定所述原始图像所对应的不确定度,包括:
获取每个分割结果中每个像素点所对应的灰度值;
基于所述分割结果集合,根据所述每个分割结果中每个像素点所对应的灰度值,确定每个像素点的方差;
根据所述每个像素点的方差以及像素点总数,确定所述原始图像所对应的不确定度。
6.根据权利要求1所述的方法,其特征在于,所述根据所述分割结果集合确定所述原始图像所对应的不确定度之后,所述方法还包括:
判断所述不确定度是否大于或等于不确定度限;
若所述不确定度大于或等于所述不确定度门限,则将所述原始图像确定为所述待标注图像。
7.根据权利要求1所述的方法,其特征在于,所述获取原始图像以及自编码器集合之前,所述方法还包括:
获取待训练图像,其中,所述待训练图像为已经过标注的图像;
通过待训练自编码器的编码器对所述待训练图像进行编码,得到编码结果;
通过所述待训练自编码器的解码器对所述编码结果进行解码,得到预测图像;
采用所述待训练图像与所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
根据所述自编码器模型参数生成自编码器,其中,所述自编码器为所述自编码器集合中的一个自编码器。
8.根据权利要求1所述的方法,其特征在于,所述通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合之前,所述方法还包括:
获取待训练图像,其中,所述待训练图像为已经过标注的图像;
通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
采用所述待训练图像与所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
根据所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络为所述图像分割网络中的一个子图像分割网络。
9.一种模型训练的方法,其特征在于,包括:
获取待训练图像,其中,所述待训练图像为已经过标注的图像;
通过待训练自编码器的编码器对所述待训练图像进行编码,得到编码结果;
通过所述待训练自编码器的解码器对所述编码结果进行解码,得到预测图像;
采用所述待训练图像与所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
根据所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
10.一种模型训练的方法,其特征在于,包括:
获取待训练图像,其中,所述待训练图像为已经过标注的图像;
通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
采用所述待训练图像与所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
根据所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
11.一种待标注图像确定装置,其特征在于,包括:
获取模,用于获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
所述获取模块,还用于通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
所述获取模块,还用于通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
确定模块,用于根据所述获取模块获取的所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
12.一种模型训练装置,其特征在于,包括:
获取模块,用于获取待训练图像,其中,所述待训练图像为已经过标注的图像;
编码模块,用于通过待训练自编码器的编码器对所述获取模块获取的所述待训练图像进行编码,得到编码结果;
解码模块,用于通过所述待训练自编码器的解码器,对所述编码模块编码后的所述编码结果进行解码,得到预测图像;
训练模块,用于采用所述待训练图像与所述解码模块解码得到的所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
生成模块,用于根据所述训练模块训练得到的所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
13.一种模型训练装置,其特征在于,包括:
获取模块,用于获取待训练图像,其中,所述待训练图像为已经过标注的图像;
所述获取模块,还用于通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
训练模块,用于采用所述待训练图像与所述获取模块获取的所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
生成模块,用于根据所述训练模块训练得到的所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
14.一种服务器,其特征在于,包括:存储器、收发器、处理器以及总线系统;
其中,所述存储器用于存储程序;
所述处理器用于执行所述存储器中的程序,包括如下步骤:
获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
根据所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像;
所述总线系统用于连接所述存储器以及所述处理器,以使所述存储器以及所述处理器进行通信。
15.一种计算机可读存储介质,包括指令,当其在计算机上运行时,使得计算机执行如权利要求1至8中任一项所述的方法,或,执行如权利要求9所述的方法,或,执行如权利要求
10所述的方法。

说明书全文

一种待标注图像确定的方法、模型训练的方法及装置

技术领域

[0001] 本申请涉及人工智能领域,尤其涉及一种待标注图像确定的方法、模型训练的方法及装置。

背景技术

[0002] 医学图像是指为了医疗或医学研究,对人体或人体某部分,以非侵入方式取得的内部组织图像,是用于辅助临床诊断的一个重要手段和参考因素,不同病症的内在异质性也在其成像表型中体现。因此,通过医学图像进行病因诊断或病灶分割,是医学图像分析中最具挑战性的任务。
[0003] 深度卷积神经网络算法已经广泛应用于病灶的分类与分割,经典的分割分类方法大多需要高精度的标注,因此,需要大量的时间进行标注。目前,提出一种在分割任务中加入随机失活(dropout)层的方法,可以得到分割结果和多种不确定度,对不确定度高的图像进行标注即可。
[0004] 然而,基于dropout机制提取不确定度存在这样的问题,由于dropout机制是基于相似结构的网络和同样的输入图像得到的,每次使用不同的节点,但是仍有大量节点的参数的一样的,因此,输出的结果还存在较高的相关性,难以将图像处理为差异巨大的结果,导致不确定度的计算缺乏合理性,从而不利于选择准确的待标注图像。发明内容
[0005] 本申请实施例提供了一种待标注图像确定的方法、模型训练的方法及装置,通过不同的自编码器能够侧重提取原始图像不同方面的信息,分化出不同自编码器对原始图像的理解,且不同的自编码器输出结果应用在不同的子图像分割网络上,使得输出的结果变化更多,从而提升不确定度的计算合理性,进而有助于选择更加准确的待标注图像。
[0006] 有鉴于此,本申请第一方面提供一种待标注图像确定的方法,包括:
[0007] 获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
[0008] 通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
[0009] 通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
[0010] 根据所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
[0011] 本申请第二方面提供一种模型训练的方法,包括:
[0012] 获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0013] 通过待训练自编码器的编码器对所述待训练图像进行编码,得到编码结果;
[0014] 通过所述待训练自编码器的解码器对所述编码结果进行解码,得到预测图像;
[0015] 采用所述待训练图像与所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0016] 根据所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0017] 本申请第三方面提供一种模型训练的方法,包括:
[0018] 获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0019] 通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0020] 采用所述待训练图像与所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0021] 根据所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0022] 本申请第四方面提供一种待标注图像确定装置,包括:
[0023] 获取模,用于获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
[0024] 所述获取模块,还用于通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
[0025] 所述获取模块,还用于通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
[0026] 确定模块,用于根据所述获取模块获取的所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
[0027] 在一种可能的设计中,在本申请实施例的第四方面的第一种实现方式中,[0028] 所述获取模块,具体用于通过第一自编码器获取所述原始图像所对应的第一编码图像;
[0029] 通过第二自编码器获取所述原始图像所对应的第二编码图像,其中,所述第二自编码器与所述第一自编码器属于两种不同类型的自编码器,所述第二编码图像和所述第一编码图像属于两个不同的图像。
[0030] 在一种可能的设计中,在本申请实施例的第四方面的第二种实现方式中,[0031] 所述获取模块,具体用于通过第一子图像分割网络获取第一分割结果,其中,所述第一分割结果包括(N+1)个第一分割子结果,所述第一分割子结果与所述编码图像和所述原始图像具有对应关系;
[0032] 通过第二子图像分割网络获取第二分割结果,其中,所述第二分割结果包括(N+1)个第二分割子结果,所述第二分割子结果与所述编码图像和所述原始图像具有对应关系,所述第二子图像分割网络与所述第一子图像分割网络属于两种不同类型的子图像分割网络。
[0033] 在一种可能的设计中,在本申请实施例的第四方面的第三种实现方式中,[0034] 所述确定模块,具体用于获取所述原始图像中每个像素点所对应的灰度值;
[0035] 根据所述分割结果集合以及所述原始图像中每个像素点所对应的灰度值,确定所述原始图像中每个像素点所对应的熵集合,其中,所述熵集合包括[(N+1)*M]个熵;
[0036] 根据所述原始图像中每个像素点所对应的熵集合,计算得到所述原始图像所对应的不确定度。
[0037] 在一种可能的设计中,在本申请实施例的第四方面的第四种实现方式中,[0038] 所述确定模块,具体用于获取每个分割结果中每个像素点所对应的灰度值;
[0039] 基于所述分割结果集合,根据所述每个分割结果中每个像素点所对应的灰度值,确定每个像素点的方差;
[0040] 根据所述每个像素点的方差以及像素点总数,确定所述原始图像所对应的不确定度。
[0041] 在一种可能的设计中,在本申请实施例的第四方面的第五种实现方式中,所述待标注图像确定装置还包括判断模块;
[0042] 所述判断模块,用于在所述确定模块根据所述分割结果集合确定所述原始图像所对应的不确定度之后,判断所述不确定度是否大于或等于不确定度限;
[0043] 所述确定模块,还用于若所述判断模块判断得到所述不确定度大于或等于所述不确定度门限,则将所述原始图像确定为所述待标注图像。
[0044] 在一种可能的设计中,在本申请实施例的第四方面的第六种实现方式中,所述待标注图像确定装置还包括编码模块、解码模块、训练模块以及生成模块;
[0045] 所述获取模块,还用于获取原始图像以及自编码器集合之前,获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0046] 所述编码模块,用于通过待训练自编码器的编码器对所述获取模块获取的所述待训练图像进行编码,得到编码结果;
[0047] 所述解码模块,用于通过所述待训练自编码器的解码器,对所述编码模块编码得到的所述编码结果进行解码,得到预测图像;
[0048] 所述训练模块,用于采用所述待训练图像与所述解码模块解码得到的所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0049] 所述生成模块,用于根据所述训练模块训练得到的所述自编码器模型参数生成自编码器,其中,所述自编码器为所述自编码器集合中的一个自编码器。
[0050] 在一种可能的设计中,在本申请实施例的第四方面的第七种实现方式中,所述待标注图像确定装置还包括训练模块以及生成模块;
[0051] 所述获取模块,还用于通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合之前,获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0052] 所述获取模块,还用于通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0053] 所述训练模块,用于采用所述待训练图像与所述获取模块获取的所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0054] 所述生成模块,用于根据所述训练模块训练得到的所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络为所述图像分割网络中的一个子图像分割网络。
[0055] 本申请第五方面提供一种模型训练装置,包括:
[0056] 获取模块,用于获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0057] 编码模块,用于通过待训练自编码器的编码器对所述获取模块获取的所述待训练图像进行编码,得到编码结果;
[0058] 解码模块,用于通过所述待训练自编码器的解码器,对所述编码模块编码后的所述编码结果进行解码,得到预测图像;
[0059] 训练模块,用于采用所述待训练图像与所述解码模块解码得到的所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0060] 生成模块,用于根据所述训练模块训练得到的所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0061] 本申请第六方面提供一种模型训练装置,包括:
[0062] 获取模块,用于获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0063] 所述获取模块,还用于通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0064] 训练模块,用于采用所述待训练图像与所述获取模块获取的所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0065] 生成模块,用于根据所述训练模块训练得到的所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0066] 本申请第七方面提供一种服务器,包括:存储器、收发器、处理器以及总线系统;
[0067] 其中,所述存储器用于存储程序;
[0068] 所述处理器用于执行所述存储器中的程序,包括如下步骤:
[0069] 获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
[0070] 通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
[0071] 通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
[0072] 根据所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像;
[0073] 所述总线系统用于连接所述存储器以及所述处理器,以使所述存储器以及所述处理器进行通信。
[0074] 本申请第八方面提供一种服务器,包括:存储器、收发器、处理器以及总线系统;
[0075] 其中,所述存储器用于存储程序;
[0076] 所述处理器用于执行所述存储器中的程序,包括如下步骤:
[0077] 获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0078] 通过待训练自编码器的编码器对所述待训练图像进行编码,得到编码结果;
[0079] 通过所述待训练自编码器的解码器对所述编码结果进行解码,得到预测图像;
[0080] 采用所述待训练图像与所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0081] 根据所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像;
[0082] 所述总线系统用于连接所述存储器以及所述处理器,以使所述存储器以及所述处理器进行通信。
[0083] 本申请第九方面提供一种服务器,包括:存储器、收发器、处理器以及总线系统;
[0084] 其中,所述存储器用于存储程序;
[0085] 所述处理器用于执行所述存储器中的程序,包括如下步骤:
[0086] 获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0087] 通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0088] 采用所述待训练图像与所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0089] 根据所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像;
[0090] 所述总线系统用于连接所述存储器以及所述处理器,以使所述存储器以及所述处理器进行通信。
[0091] 本申请的第十方面提供了一种计算机可读存储介质,所述计算机可读存储介质中存储有指令,当其在计算机上运行时,使得计算机执行上述各方面所述的方法。
[0092] 从以上技术方案可以看出,本申请实施例具有以下优点:
[0093] 本申请实施例中,提供了一种待标注图像确定的方法,首先获取原始图像以及自编码器集合,原始图像为未经过标注的图像,然后通过自编码器集合获取原始图像所对应的编码图像集合,编码图像集合包括N个编码图像,编码图像与自编码器之间具有对应关系,再通过图像分割网络获取编码图像集合以及原始图像所对应的分割结果集合,图像分割网络包括M个子图像分割网络,分割结果集合包括[(N+1)*M]个分割结果,最后根据分割结果集合确定原始图像所对应的不确定度,其中,不确定度用于确定待标注图像。通过上述方式,将原始图像通过不同编码策略的自编码器进行改变,不同的自编码器直接能够侧重提取该原始图像不同方面的信息,分化出不同自编码器对原始图像的理解,使得一部分重建后相似而另一部分细节就出现差异。且不同的自编码器输出结果应用在不同的子图像分割网络上,使得输出的结果变化更多,从而提升不确定度的计算合理性,进而有助于选择更加准确的待标注图像。附图说明
[0094] 图1为本申请实施例中待标注图像选择系统的一个架构示意图;
[0095] 图2为本申请实施例中选择待标注图像的一个流程示意图;
[0096] 图3为本申请实施例中待标注图像确定的方法一个实施例示意图;
[0097] 图4为本申请实施例中计算不确定度的一个网络架构示意图;
[0098] 图5为本申请实施例中U-网络的一个结构示意图;
[0099] 图6为本申请实施例中全卷积网络的一个结构示意图;
[0100] 图7为本申请实施例中训练自编码器的一个实施例示意图;
[0101] 图8为本申请实施例中自编码器的一个原理示意图;
[0102] 图9为本申请实施例中训练子图像分割网络的一个实施例示意图;
[0103] 图10为本申请实施例中基于子图像分割网络的一个原理示意图;
[0104] 图11为本申请实施例中模型训练的方法一个实施例示意图;
[0105] 图12为本申请实施例中模型训练的方法另一个实施例示意图;
[0106] 图13为本申请实施例中待标注图像选择装置的一个实施例示意图;
[0107] 图14为本申请实施例中待标注图像选择装置的另一个实施例示意图;
[0108] 图15为本申请实施例中待标注图像选择装置的另一个实施例示意图;
[0109] 图16为本申请实施例中待标注图像选择装置的另一个实施例示意图;
[0110] 图17为本申请实施例中模型训练装置的一个实施例示意图;
[0111] 图18为本申请实施例中模型训练装置的另一个实施例示意图;
[0112] 图19为本申请实施例中服务器的一个结构示意图;
[0113] 图20为本申请实施例中终端设备的一个结构示意图。

具体实施方式

[0114] 本申请实施例提供了一种待标注图像确定的方法、模型训练的方法及装置,通过不同的自编码器能够侧重提取原始图像不同方面的信息,分化出不同自编码器对原始图像的理解,且不同的自编码器输出结果应用在不同的子图像分割网络上,使得输出的结果变化更多,从而提升不确定度的计算合理性,进而有助于选择更加准确的待标注图像。
[0115] 本申请的说明书权利要求书及上述附图中的术语“第一”、“第二”、“第三”、“第四”等(如果存在)是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便这里描述的本申请的实施例例如能够以除了在这里图示或描述的那些以外的顺序实施。此外,术语“包括”和“对应于”以及他们的任何变形,意图在于覆盖不排他的包含,例如,包含了一系列步骤或单元的过程、方法、系统、产品或设备不必限于清楚地列出的那些步骤或单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它步骤或单元。
[0116] 应理解,本申请提供的待标注图像选择方法具有较大的应用前景,可以通过利用小数据集学习,并识别数据的不确定度,然后帮助用户选择对深度卷积神经网络更加有意义的数据,能够增加训练集中没见过的数据,再对这些数据进行标注即可。从而使得在较小标准量的情况下取得较大的训练和测试精度。
[0117] 具体地,本申请基于人工智能(Artificial Intelligence,AI)的技术对图像进行不确定度的计算,从而利用不确定度对待标注图像进行选择。AI是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能,感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。换句话说,人工智能是计算机科学的一个综合技术,它企图了解智能的实质,并生产出一种新的能以人类智能相似的方式做出反应的智能机器。人工智能也就是研究各种智能机器的设计原理与实现方法,使机器具有感知、推理与决策的功能。
[0118] 人工智能技术是一门综合学科,涉及领域广泛,既有硬件层面的技术也有软件层面的技术。人工智能基础技术一般包括如传感器、专用人工智能芯片计算、分布式存储、大数据处理技术、操作/交互系统、机电一体化等技术。人工智能软件技术主要包括计算机视觉技术、语音处理技术、自然语言处理技术以及机器学习/深度学习等几大方向。
[0119] 本申请采用机器学习(Machine Learning,ML)的方式对自编码器和图像分割模型训练训练。ML是一门多领域交叉学科,涉及概率论、统计学、逼近论、凸分析、算法复杂度理论等多门学科。专门研究计算机怎样模拟或实现人类的学习行为,以获取新的知识或技能,重新组织已有的知识结构使之不断改善自身的性能。机器学习是人工智能的核心,是使计算机具有智能的根本途径,其应用遍及人工智能的各个领域。机器学习和深度学习通常包括人工神经网络、置信网络、强化学习、迁移学习、归纳学习、式教学习等技术。
[0120] 为了便于理解,本申请提出了一种待标注图像确定的方法,该方法应用于图1所示的待标注图像选择系统,请参阅图1,图1为本申请实施例中待标注图像选择系统的一个架构示意图,如图所示,终端设备获取大量的医学图像,这些医学图像包含但不仅限于电子计算机断层扫描(Computed Tomography,CT)图像、磁共振成像(Magnetic Resonance Imaging,MRI)图像、超声(B超、彩色多普勒超声、心脏彩超以及三维彩超)图像、X光图像、心电图脑电图以及光学摄影等由医学仪器生成的图像。医学仪器包含但不仅限于X光成像仪器、CT、MRI、心电图仪器以及脑电图仪器等。医学图像是用于辅助临床诊断的一种重要手段和参考因素,不同病征的内在异质性也在其成像表型(外观和形状)中体现。医学仪器采集到的医学图像可以存放于终端设备。
[0121] 需要说明的是,终端设备包含但不仅限于平板电脑笔记本电脑、掌上电脑、手机、语音交互设备及个人电脑(personal computer,PC),此处不做限定。
[0122] 通过终端设备可以看到海量的医学图像,如果对这些医学图像都进行标注,需要耗费大量时间,因此,需要从海量的医学图像中筛选不确定度较高的医学图像进行标注。于是,终端设备将大量的医学图像发送至服务器,由服务器采用多个自编码器和多个图像分割模型计算出医学图像的不确定度,对于不确定度较高的医学图像进行标注即可。
[0123] 请参阅图2,图2为本申请实施例中选择待标注图像的一个流程示意图,如图所示,具体地,在模型训练的过程中,前端A可以接收有标签的医学图像和无标签的医学图像,经过预处理算法之后将这些医学图像上传至后端。可以理解的是,预处理算法包含但不仅限于平移、旋转和对称等数据增广算法。后端使用经过处理后的有标签医学图像进行训练,对无标签医学图像进行不确定度计算,然后输出不确定度到前端B,医生可以通过前端B选取不确定度较高的数据进行标注。
[0124] 其中,后端可以是指一台服务器或服务器集群,在预测的过程中,后端也可以是另一台具有计算能的终端设备,比如电脑。前端A可以是医学仪器,也可以是存储有医学图像的终端设备,比如电脑。前端B可以是终端设备,比如电脑。
[0125] 结合上述介绍,下面将对本申请中待标注图像确定的方法进行介绍,请参阅图3,本申请实施例提供的待标注图像确定的方法涉及人工智能的图像处理等技术,具体通过如下实施例进行说明:
[0126] 101、获取原始图像以及自编码器集合,其中,原始图像为未经过标注的图像,自编码器集合包括N个自编码器,N为大于或大于1的整数;
[0127] 本实施例中,待标注图像选择装置获取原始图像以及自编码器集合,可以理解的是,待标注图像选择装置通常部署于服务器,或者可以部署于服务器集群,还可以部署于具有较强计算能力的终端设备上,本申请以部署于在服务器为例进行介绍,然而这不应理解为对本申请的限定。
[0128] 其中,原始图像可以是医学图像,包含但不仅限于CT图像、MRI图像、超声图像、X光图像、心电图以及脑电图。也可以是其他类型的图像,比如监控图像,本申请以医学图像为例进行介绍,然而这不应理解为对本申请的限定。
[0129] 自编码器(autoencoder,AE)是一类在半监督学习和非监督学习中使用的人工神经网络(Artificial Neural Networks,ANNs),其功能是通过将输入信息作为学习目标,对输入信息进行表征学习(representation learning)。
[0130] 自编码器包含编码器(encoder)和解码器(decoder)两部分。按学习范式,自编码器可以被分为收缩自编码器(undercomplete autoencoder)、正则自编码器(regularized autoencoder)和变分自编码器(Variational AutoEncoder,VAE),其中前两者是判别模型、后者是生成模型。按构筑类型,自编码器可以是前馈结构或递归结构的神经网络。自编码器具有一般意义上表征学习算法的功能,被应用于降维(dimensionality reduction)和异常值检测(anomaly detection)。包含卷积层构筑的自编码器可被应用于计算机视觉问题,包括图像降噪(image denoising)及神经格迁移(neural style transfer)等。
[0131] 102、通过自编码器集合获取原始图像所对应的编码图像集合,其中,编码图像集合包括N个编码图像,编码图像与自编码器之间具有对应关系;
[0132] 本实施例中,待标注图像选择装置通过通过自编码器集合获取原始图像所对应的编码图像集合。其中,原始图像为未经过标注的图像,自编码器集合包括N个自编码器,将原始图像分别输入到这N个自编码器,从而得到N个编码图像。为了便于理解,请参阅表1,表1为自编码器与编码图像之间的对应关系一个示意。
[0133] 表1
[0134]输入 自编码器类型 输出
原始图像 自编码器A 编码图像1
原始图像 自编码器B 编码图像2
原始图像 自编码器C 编码图像3
原始图像 自编码器D 编码图像4
原始图像 自编码器F 编码图像5
[0135] 由此可见,编码图像与自编码器之间具有对应关系,自编码器通过对输入的原始图像提取不同的重构信息,重构信息就是从不同损失函数定义的自编码器生成的图像,用自编码器通过编码——解码结构生成的与原始图像相似的重构图像信息,即得到编码图像。
[0136] 103、通过图像分割网络获取编码图像集合以及原始图像所对应的分割结果集合,其中,图像分割网络包括M个子图像分割网络,分割结果集合包括[(N+1)*M]个分割结果,M为大于或等于1的整数;
[0137] 本实施例中,待标注图像选择装置将N个编码图像以及原始图像作为图像分割网络的输入,其中,图像分割网络包括M个子图像分割网络,因此,得到的分割结果为[(N+1)*M]。为了便于理解,请参阅表2,表2为编码图像、原始图像与分割结果一个示意。
[0138] 表2
[0139]输入 子图像分割网络类型 输出
原始图像 子图像分割网络A 分割结果1
编码图像1 子图像分割网络A 分割结果2
编码图像2 子图像分割网络A 分割结果3
原始图像 子图像分割网络B 分割结果4
编码图像1 子图像分割网络B 分割结果5
编码图像2 子图像分割网络B 分割结果6
[0140] 由此可见,假设有2个编码图像,即N=2,将原始图像和2个编码图像都作为子图像分割网络A的输入,从而分别输出对应的分割结果,即得到3个分割结果。类似地,将原始图像和2个编码图像都作为子图像分割网络B的输入,从而分别输出对应的分割结果,即得到3个分割结果。从而总共得到6个分割结果,即[(N+1)*M]=[(2+1)*2]=6。
[0141] 利用多个不同无监督子图像分割网络可以提取图像的底层表达信息,从而实现不同无监督网络对确定性区域和不确定性区域的划分,
[0142] 104、根据分割结果集合确定原始图像所对应的不确定度,其中,不确定度用于确定待标注图像。
[0143] 本实施例中,待标注图像选择装置根据分割结果集合,计算得到该原始图像的不确定度。基于该不确定度可以判断该原始图像是否适合作为待标注图像,通常情况下,不确定度越大,作为待标注图像的可能性越高。
[0144] 为了便于理解,请参阅图4,图4为本申请实施例中计算不确定度的一个网络架构示意图,如图所示,假设有N个自编码器,将原始图像分别输入至不同的自编码器中,从而得到N个编码图像,再将N个编码图像和原始图像分别输入至不同的子图像分割网络,共有M个子图像分割网络,即得到[(N+1)*M]个分割结果,最后基于[(N+1)*M]个分割结果计算得到原始图像的不确定度,不确定度可以是方差或者熵,还可以是其他的表示方式。
[0145] 本申请实施例中,提供了一种待标注图像确定的方法,首先获取原始图像以及自编码器集合,原始图像为未经过标注的图像,然后通过自编码器集合获取原始图像所对应的编码图像集合,编码图像集合包括N个编码图像,编码图像与自编码器之间具有对应关系,再通过图像分割网络获取编码图像集合以及原始图像所对应的分割结果集合,图像分割网络包括M个子图像分割网络,分割结果集合包括[(N+1)*M]个分割结果,最后根据分割结果集合确定原始图像所对应的不确定度,其中,不确定度用于确定待标注图像。通过上述方式,将原始图像通过不同编码策略的自编码器进行改变,不同的自编码器直接能够侧重提取该原始图像不同方面的信息,分化出不同自编码器对原始图像的理解,使得一部分重建后相似而另一部分细节就出现差异。且不同的自编码器输出结果应用在不同的子图像分割网络上,使得输出的结果变化更多,从而提升不确定度的计算合理性,进而有助于选择更加准确的待标注图像。
[0146] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第一个可选实施例中,通过自编码器集合获取原始图像所对应的编码图像集合,可以包括:
[0147] 通过第一自编码器获取原始图像所对应的第一编码图像;
[0148] 通过第二自编码器获取原始图像所对应的第二编码图像,其中,第二自编码器与第一自编码器属于两种不同类型的自编码器,第二编码图像和第一编码图像属于两个不同的图像。
[0149] 本实施例中,介绍了一种基于多个自编码器生成编码图像的方法。待标注图像选择装置可以将原始图像输入至第一自编码器,由第一自编码器输出第一编码图像,并且将原始图像输入至第二自编码器,由第二自编码器输出第二编码图像。可以理解的是,第一自编码器和第二自编码器仅为一个示意,在实际应用中,还存在更多数量和更多类型自编码器。
[0150] 具体地,自编码器类型包含但不仅限于稀疏自编码器、变分自编码器(Variational Auto-Encoder,VAE)、收缩自编码器(Contractive Auto-Encoder,CAE)以及降噪自编码器(Denoising Auto-Encoder,DAE),且自编码器的数量可以是大于或等于1的整数。本申请中的自编码器集合可以是2个稀疏自编码器1、2个稀疏自编码器2、2个VAE、2个CAE以及2个降噪自编码器,共10个自编码器。
[0151] 其中,稀疏自编码器可以自动从无标注数据中学习特征,可以给出比原始数据更好的特征描述。在实际运用时可以用稀疏编码器发现的特征取代原始数据,这样往往能带来更好的结果。稀疏自编码器1和稀疏自编码器2是两种不同类型的稀疏自编码器。
[0152] 变分自编码器是一类重要的生成模型,相比于自编码器,VAE更倾向于数据生成。只要训练好了解码器,就可以从标准正态分布生成数据作为解码器的输入,来生成类似但不同于训练数据的新样本。
[0153] 收缩自编码器是只在局部收缩,目标是学习数据的流形结构。CAE的训练目标函数是重构误差和收缩罚项(即正则项)的总和,通过最小化该目标函数使已学习到的表示C(x)尽量对输入x保持不变。它的惩罚项Ω(h)是平方Frobenius范数(元素平方的和),作用于与编码器的函数相关偏导数的Jacobian矩阵。简单来说,它的惩罚项是编码器输出的特征向量的元素平方的和。这个处罚项迫使模型学习一个在x变化小时目标也没有太大变化的函数。因为这个惩罚只对训练数据适用,它迫使自编码器学习可以反映训练数据分布信息的特征。
[0154] 降噪自编码器可以接受加噪的输入来进行训练,核心思想是,一个能够从中恢复出原始信号的神经网络表达未必是最好的,能够对“损坏”的原始数据编码和解码,然后还能恢复真正的原始数据,这样的特征才是好的。
[0155] 基于上述介绍,在主干网络之前,会将输入通过不同策略的自编码器改变,不同的自编码器直接能够侧重提取该图像不同方面的信息,分化出不同自编码器对该图像的理解,即一部分重建后相似而另一部分细节就出现差异。不同的自编码器也可以加入不同的网络结构单元,使得自编码器更加多元,从而输入的不确定度增大。采用不同的自编码器输出结果应用在不同的子图像分割网络上,从而使得输出图像变化更多,不确定度也能计算的更加精确。
[0156] 其次,本申请实施例中,提供了一种基于多个自编码器生成编码图像的方法,即通过第一自编码器获取原始图像所对应的第一编码图像,并且通过第二自编码器获取原始图像所对应的第二编码图像,第二自编码器与第一自编码器属不同类型的自编码器。通过上述方式,能够利用多个自编码器对图像进行编码,由于不同自编码器能侧重提取图像的不同特征进行编码,利用多个自编码器生成的不同图像在主干位置变换不大而细节变化较大,由此,可以将网络理解的细节定义为“不确定度”,这部分就是自编码器前后改变较大的区域,从而分割时会造成不确定度大的区域分割结果差异也较大,提升不确定度计算的可靠性。
[0157] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第二个可选实施例中,通过图像分割网络获取编码图像集合以及原始图像所对应的分割结果集合,包括:
[0158] 通过第一子图像分割网络获取第一分割结果,其中,第一分割结果包括(N+1)个第一分割子结果,第一分割子结果与编码图像和原始图像具有对应关系;
[0159] 通过第二子图像分割网络获取第二分割结果,其中,第二分割结果包括(N+1)个第二分割子结果,第二分割子结果与编码图像和原始图像具有对应关系,第二子图像分割网络与第一子图像分割网络属于两种不同类型的子图像分割网络。
[0160] 本实施例中,介绍了一种基于多个子图像分割网络生成分割结果的方法。待标注图像选择装置可以将N个编码图像和原始图像输入至第一子图像分割网络,得到(N+1)个第一分割子结果,比如1个编码图像对应一个第一分割子结果,原始图像对应另一个第一分割子结果,最后得到(N+1)个第一分割子结果,这(N+1)个第一分割子结果几位第一分割结果。类似地,将N个编码图像和原始图像输入至第二子图像分割网络,得到(N+1)个第二分割子结果,比如1个编码图像对应一个第二分割子结果,原始图像对应另一个第二分割子结果,最后得到(N+1)个第二分割子结果,这(N+1)个第二分割子结果几位第二分割结果。
[0161] 可以理解的是,第一子图像分割网络和第二子图像分割网络仅为一个示意,在实际应用中,还存在更多数量和更多类型子图像分割网络。具体地,子图像分割网络类型包含但不仅限于U-网络(U-Net)以及全卷积网络(Fully Convolutional Networks,FCN)。
[0162] 为了便于说明,请参阅图5,图5为本申请实施例中U-网络的一个结构示意图,如图所示,U-Net是卷积神经网络的一种变形,整个神经网络主要有两部分组成,分别为收缩路径和扩展路径。搜索路径主要是用来捕捉图片中的上下文信息(context information),而与之相对称的扩展路径则是为了对图片中所需要分割出来的部分进行精准定位(localization)。U-Net诞生的一个主要前提是,很多时候深度学习的结构需要大量的sample和计算资源,但是U-Net基于全卷积神经网络(Fully Convultional Neural Network,FCN)进行改进,并且利用数据增强可以对一些比较少样本的数据进行训练,特别是医学方面相关的数据,这是因为医学数据比一般我们所看到的图片及其他文本数据的获取成本更大,不论是时间还是资源的消耗,所以U-Net的出现对于深度学习用于较少样本的医学图像是很有帮助的。
[0163] 请参阅图6,图6为本申请实施例中全卷积网络的一个结构示意图,如图所示,FCN将传统卷积神经网络(Convolutional Neural Networks,CNN)中的全连接层转化成一个个的卷积层。图6展示了卷积和反卷积上采样的过程,是对第5层的输出(32倍放大)反卷积到原图大小,得到的结果还是不够精确,一些细节无法恢复。于是将第4层的输出和第3层的输出也依次反卷积,分别需要16倍和8倍上采样。
[0164] 其次,本申请实施例中,提供了一种基于多个子图像分割网络生成分割结果的方法,即通过第一子图像分割网络获取第一分割结果,通过第二子图像分割网络获取第二分割结果,其中,第二子图像分割网络与第一子图像分割网络属于不同类型的子图像分割网络。通过上述方式,能够利用多个子图像分割网络对图像进行分割,由于不同子图像分割网络能侧重提取图像的分割区域,采用不同的自编码器输出结果应用在不同的子图像分割网络上,从而使输出图像变化更多,不确定度的结果也更加精确。
[0165] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第三个可选实施例中,根据分割结果集合确定原始图像所对应的不确定度,可以包括:
[0166] 获取原始图像中每个像素点所对应的灰度值;
[0167] 根据分割结果集合以及原始图像中每个像素点所对应的灰度值,确定原始图像中每个像素点所对应的熵集合,其中,熵集合包括[(N+1)*M]个熵;
[0168] 根据原始图像中每个像素点所对应的熵集合,计算得到原始图像所对应的不确定度。
[0169] 本实施例中,提供了一种基于熵的计算得到不确定度的方式,在得到[(N+1)*M]个分割结果计算得到原始图像的不确定度。具体地,假设N为10,M为2,则可以得到(10+1)*2=22个分割结果。对这22个分割结果计算一个不确定度。具体可以采用如下方式进行计算:
[0170]
[0171] 其中,H[xi]表示第i个像素点的灰度值所对应的熵,xi表示原始图像中第i个像素点的灰度值,T表示分割结果的总数,这里T可以为22,t表示大于或等于1,且小于或等于T的整数。C表示网络分割模型的总数,这里C可以为2,c表示大于或等于1,且小于或等于C的整数。 表示第t个分割结果的预测概率。
[0172] 由此得到原始图像中每个像素点所对应的熵集合,即得到“熵图”,假设每个像素点22个结果,用这22个结果可以得到一个熵的值,拼起来就是一张图。再对所有点的熵取平均值即为不确定度。经过实验,利用无标注数据不确定度的排序,取无标签数据中不确定度较大的前20%数据标注和原本有标签的数据一起训练,得到的dice值与所有无标签数据标注后与有标签数据一起训练的结果接近,远远高于无标签数据中随机取20%数据标注和原本有标签的数据一起训练。
[0173] 再次,本申请实施例中,提供了一种计算不确定度的方式,即获取原始图像中每个像素点所对应的灰度值,然后根据分割结果集合以及原始图像中每个像素点所对应的灰度值,确定原始图像中每个像素点所对应的熵集合,最后根据原始图像中每个像素点所对应的熵集合,计算得到原始图像所对应的不确定度。通过上述方式,为不确定度的计算提供了可行的实现方式,从而提升方案的可操作性和可靠性。
[0174] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第四个可选实施例中,根据分割结果集合确定原始图像所对应的不确定度,可以包括:
[0175] 获取每个分割结果中每个像素点所对应的灰度值;
[0176] 基于分割结果集合,根据每个分割结果中每个像素点所对应的灰度值,确定每个像素点的方差;
[0177] 根据每个像素点的方差以及像素点总数,确定原始图像所对应的不确定度。
[0178] 本实施例中,提供了一种基于方差的计算得到不确定度的方式,在得到[(N+1)*M]个分割结果计算得到原始图像的不确定度。具体地,假设N为10,M为2,则可以得到(10+1)*2=22个分割结果。对这22个分割结果计算一个不确定度。具体可以采用如下方式进行计算:
[0179]
[0180]
[0181] 其中,AVG表示平均值,DEV表示方差,xi表示原始图像中第i个像素点的灰度值,T表示分割结果的总数,这里T可以为22,t表示大于或等于1,且小于或等于T的整数。C表示网络分割模型的总数,这里C可以为2,c表示大于或等于1,且小于或等于C的整数。 表示第t个分割结果的预测概率。
[0182] 即,方差就是基于每个像素点的[(N+1)*M]个结果得到的方差,整张图的方差加起来除以像素数量即为不确定度。
[0183] 可以理解的是,本申请还可以采用交叉熵的方式计算不确定度,此次不做限定。
[0184] 再次,本申请实施例中,提供了另一种计算不确定度的方式,即首先获取每个分割结果中每个像素点所对应的灰度值,然后基于分割结果集合,根据每个分割结果中每个像素点所对应的灰度值,确定每个像素点的方差,最后根据每个像素点的方差以及像素点总数,确定原始图像所对应的不确定度。通过上述方式,为不确定度的计算提供了另一种可行的实现方式,从而提升方案的可操作性、可靠性以及灵活性。
[0185] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第五个可选实施例中,根据分割结果集合确定原始图像所对应的不确定度之后,还可以包括:
[0186] 判断不确定度是否大于或等于不确定度门限;
[0187] 若不确定度大于或等于不确定度门限,则将原始图像确定为待标注图像。
[0188] 本实施例中,介绍了一种待标注图像选择装置自动选择待标注图像的方式。在待标注图像选择装置计算得到原始图像的不确定度之后,可以判断该不确定度是否大于或等于不确定度门限。具体可以采用如下四种方式实现:
[0189] 一、固定不确定度门限;
[0190] 假设不确定度的取值为大于或等于0,且小于或等于1,那么在0至1这个范围内预先设置一个不确定度门限,假设为0.8。于是,在计算得到原始图像的不确定度之后,判断该不确定度是否大于或等于0.8,如果是,则表示该原始图像是需要进行标注的待标注图像,此时,待标注图像选择装置可以直接将待标注图像推送到终端设备上,从而使得用户直接在该终端设备上对待标注图像进行标注即可。
[0191] 二、固定选取数据的百分比;
[0192] 假设不确定度的取值为大于或等于0,且小于或等于1,此时需要从0至1这个范围内选择一个不确定度门限。该不确定度门限的确定方式为,假设一共预测了1000个图像,将这1000个图像的不确定度从大至小排序,然后选择出不确定度排序最靠前的10%的图像作为待标注图像,即选择出不确定度靠前的100个图像作为待标注图像。此时,待标注图像选择装置可以直接将待标注图像推送到终端设备上,从而使得用户直接在该终端设备上对待标注图像进行标注即可。
[0193] 三、固定选取数据量;
[0194] 假设不确定度的取值为大于或等于0,且小于或等于1,此时需要从0至1这个范围内选择一个不确定度门限。该不确定度门限的确定方式为,假设一共预测了1000个图像,将这1000个图像的不确定度从大至小排序,然后选择不确定度排序最靠前的50个图像作为待标注图像。此时,待标注图像选择装置可以直接将待标注图像推送到终端设备上,从而使得用户直接在该终端设备上对待标注图像进行标注即可。
[0195] 四、浮动不确定门限;
[0196] 假设不确定度的取值为大于或等于0,且小于或等于1,此时需要从0至1这个范围内选择一个不确定度门限。该不确定度门限的确定方式为,假设一共预测了1000个图像,将这1000个图像的不确定度从大至小排序,然后取出不确定度最大的前100个图像,再计算这100个图像的不确定度平均值,假设得到0.85,于是也将0.85确定为不确定度门限。在计算得到原始图像的不确定度之后,判断该不确定度是否大于或等于0.85,如果是,则表示该原始图像是需要进行标注的待标注图像,此时,待标注图像选择装置可以直接将待标注图像推送到终端设备上,从而使得用户直接在该终端设备上对待标注图像进行标注即可。
[0197] 其次,本申请实施例中,提供了一种基于不确定度自行提供待标注图像的方法,在根据分割结果集合确定原始图像所对应的不确定度之后,还可以判断不确定度是否大于或等于不确定度门限,若不确定度大于或等于不确定度门限,则将原始图像确定为待标注图像。通过上述方式,能够根据不确定度自动判断该原始图像是否适合作为待标注图像,这样的话,一方面能够提升待标注图像选择的效率,无需人为根据不确定度进行待标注图像的筛选,另一方面,能够有效地提升待标注图像的选择的准确度,防止出现待标注图像漏选的情况。
[0198] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第六个可选实施例中,获取原始图像以及自编码器集合之前,还可以包括:
[0199] 获取待训练图像,其中,待训练图像为已经过标注的图像;
[0200] 通过待训练自编码器的编码器对待训练图像进行编码,得到编码结果;
[0201] 通过待训练自编码器的解码器对编码结果进行解码,得到预测图像;
[0202] 采用待训练图像与预测图像对待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0203] 根据自编码器模型参数生成自编码器,其中,自编码器为自编码器集合中的一个自编码器。
[0204] 本实施例中,介绍了一种训练自编码器的方式。在训练的过程中,需要使用待标签的数据,即对待训练图像进行训练,请参阅图7,图7为本申请实施例中训练自编码器的一个实施例示意图,如图所示,通过大量的待训练图像可以分别训练得到自编码器1至自编码器N。其中,该待训练图像可以是经过处理处理后的图像,比如对待训练图像进行归一化处理,对于没有固定的形状大小和方向性的病灶区域,可以对该待训练图像进行翻转、旋转、放缩和对比度增强等数据增广的操作,从而增加训练样本数量,增大方向性和不同大小规模下的信息价值。
[0205] 请参阅图8,图8为本申请实施例中自编码器的一个原理示意图,如图所示,自编码器包括编码器和解码器两个部分,编码器用于将输入压缩成潜在空间表征,解码器用于重构来自潜在空间表征的输入。其中,编码器的输入表示为X,经过编码后得到的编码结果表示为z,即z=h(X)。解码器的输入为z,输出为X’,即X’=y(X)。本申请需要针对每个针对自编码器都进行训练,采用待训练的自编码器对待训练图像进行编码和解码之后得到预测图像,当预测图像和待训练图像之间的差距足够小时,损失函数达到收敛。此时,取出编码器和解码器的自编码器模型参数,根据该自编码器模型参数生成最终的自编码器。
[0206] 具体地,以1号稀疏自编码器为例,KL散度为 均方误差(Mean Square Error,MSE)为 其中,β=3,ρ=0.2或者ρ=0.05。损失函数可以为[0207] 以2号稀疏自编码器为例,L1为 MES为 其中,λ=1,L1=
10e-3或者L1=10e-5。损失函数可以为
[0208] 以变分自编码器为例,KL散度为λ(1+logσ2-u2-exp(logσ2)),MES为其中,λ=0.5或λ=0.2。损失函数可以为
[0209] 以收缩自编码器为例,L2为 MES为 其中,λ=1,L2=10e-3或L2=10e-5。损失函数为
[0210] 以收缩自编码器为例,MES为 其中,高斯噪声参数为Theta=0.5或Theta=0.2。损失函数为
[0211] 可以理解的是,上述各类自编码器类型仅为一个示意,不应理解为对本申请的限定,此外,上述各个自编码器的损失函数也为一个示意,在实际应用中,还可以根据情况对损失函数进行灵活调整。
[0212] 其次,本申请实施例中,提供了一种自编码器的训练方法,即首先获取待训练图像,然后通过待训练自编码器的编码器对待训练图像进行编码,得到编码结果,再通过待训练自编码器的解码器对编码结果进行解码,得到预测图像在,最后采用待训练图像与预测图像对待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数,利用自编码器模型参数生成自编码器。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0213] 可选地,在上述图3以及图3对应的各个实施例的基础上,本申请实施例提供的待标注图像确定的方法第七个可选实施例中,通过图像分割网络获取编码图像集合以及原始图像所对应的分割结果集合之前,还可以包括:
[0214] 获取待训练图像,其中,待训练图像为已经过标注的图像;
[0215] 通过待训练子图像分割网络获取待训练图像所对应的预测图像;
[0216] 采用待训练图像与预测图像对待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0217] 根据网络模型参数生成子图像分割网络,其中,子图像分割网络为图像分割网络中的一个子图像分割网络。
[0218] 本实施例中,介绍一种训练子图像分割网络的方式。在训练的过程中,需要使用待标签的数据,即对待训练图像进行训练,请参阅图9,图9为本申请实施例中训练子图像分割网络的一个实施例示意图,如图所示,通过大量的待训练图像可以分别训练得到子图像分割网络1至子图像分割网络M。其中,该待训练图像可以是经过处理处理后的图像,比如对待训练图像进行归一化处理,对于没有固定的形状大小和方向性的病灶区域,可以对该待训练图像进行翻转、旋转、放缩和对比度增强等数据增广的操作,从而增加训练样本数量,增大方向性和不同大小规模下的信息价值。
[0219] 具体地,将有标签的待训练图像分为训练集和测试集,可以分别用经典的U-Net和FCN训练两个子图像分割网络,基于自适应矩估计(adaptive moment estimation,Adam)的梯度下降法更新网络的参数,初始学习率为0.05,Adam中的betas=(0.95,0.9995)。经过子图像分割网络得到概率值,分割为一个概率图像,每一个像素值为一个0到1的概率,损失函数为dice值,通过最小化损失函数,我们可以计算误差梯度并通过反向传播更新网络的梯度。在上述网络训练完成之后,利用最终的预测概率值,得到分割结果,
[0220] 下面将介绍基于交叉熵训练子图像分割网络的方法。为了便于理解,请参阅图10,图10为本申请实施例中基于子图像分割网络的一个原理示意图,具体地,图像分割中最常用的损失函数是逐像素交叉熵损失。该损失函数分别检查每个像素,将类预测(深度方向的像素向量)与热编码目标向量进行比较。假设灰度值大于0.5的像素点记为1,小于或等于0.5的像素点记为0。0.83大于0.5,记为1,其他的像素点灰度值都小于0.5,因此都记为0。
[0221] 由此可见,交叉熵的损失函数单独评估每个像素矢量的类预测,然后对所有像素求平均值,所以可以认为图像中的像素被平等的学习了。但是,医学图像中常出现类别不均衡的问题,由此导致训练会被像素较多的类主导,对于较小的物体很难学习到其特征,从而降低网络的有效性。
[0222] 下面将介绍基于dice值训练子图像分割网络的方法。采用dice值衡量两个样本的重叠部分。该指标范围从0到1,其中“1”表示完整的重叠。为了计算预测的分割图的dice值,将重叠部分近似为预测图和标签之间的点乘,并将结果函数中的元素相加。
[0223] 其次,本申请实施例中,提供了一种子图像分割网络的训练方法,即首先获取待训练图像,然后通过待训练子图像分割网络获取待训练图像所对应的预测图像,再采用待训练图像与预测图像对待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数,最后根据网络模型参数生成子图像分割网络。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0224] 结合上述介绍,下面将基于自编码器的训练方式,对本申请提供的模型训练方法进行介绍,请参阅图11,本申请实施例中模型训练的方法一个实施例包括:
[0225] 201、获取待训练图像,其中,待训练图像为已经过标注的图像;
[0226] 本实施例中,模型训练装置获取待训练图像,可以理解的是,待训练图像是已经经过标注的图像,且训练的过程中需要采用大量的待训练图像进行训练,为了便于说明,这里将以对一个待训练图像为例进行介绍。模型训练装置通常部署于服务器。
[0227] 待训练图像可以是包括四个模态的BraTS18比赛用MR数据集,四个模态分别为T1、T2、T1C和FLAIR。当然,在实际应用中还可以采用其他类型的待训练图像。接下来,需要对待训练图像进行归一化处理,对于没有固定的形状大小和方向性的病灶区域,可以对该待训练图像进行翻转、旋转、放缩和对比度增强等数据增广的操作,从而增加训练样本数量,增大方向性和不同大小规模下的信息价值。
[0228] 202、通过待训练自编码器的编码器对待训练图像进行编码,得到编码结果;
[0229] 本实施例中,模型训练装置将待训练图像输入至待训练自编码器的编码器,编码器用于将输入压缩成潜在空间表征。经过待训练自编码器的编码器之后,得到编码结果。
[0230] 203、通过待训练自编码器的解码器对编码结果进行解码,得到预测图像;
[0231] 本实施例中,模型训练装置将编码得到的编码结果输入到待训练自编码器的解码器,解码器用于重构来自潜在空间表征的输入。经过待训练自编码器的接码器之后,得到预测图像。可以理解的是,待训练自编码器的编码器和解码器初始值可以是预先设定的。
[0232] 204、采用待训练图像与预测图像对待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0233] 本实施例中,模型训练装置基于待训练图像和预测图像,采用损失函数进行计算。具体地,不同类型的待训练自编码器往往采用不同的损失函数,可参与上述实施例的介绍,此次不做赘述。当损失函数收敛时,获取对应的自编码器模型参数。
[0234] 205、根据自编码器模型参数生成自编码器,其中,自编码器用于对原始图像进行编码,得到编码图像,编码图像用于通过子图像分割网络生成分割结果,分割结果用于确定原始图像的不确定度,不确定度用于确定待标注图像,原始图像为未经过标注的图像。
[0235] 本实施例中,模型训练装置根据生成的自编码器参数,将该自编码器参数作为自编码器的模型参数,从而得到最终的自编码器。于是,在预测的过程中,可以使用该自编码器对原始图像进行编码。
[0236] 可以理解的是,在实际应用中,需要对多个自编码器都执行如步骤201至步骤205的操作,这里仅以训练一个自编码器为例进行介绍,然而这不应理解为对本申请的限定。
[0237] 本申请实施例中,提供了一种自编码器的训练方法,即首先获取待训练图像,然后通过待训练自编码器的编码器对待训练图像进行编码,得到编码结果,再通过待训练自编码器的解码器对编码结果进行解码,得到预测图像在,最后采用待训练图像与预测图像对待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数,利用自编码器模型参数生成自编码器。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0238] 结合上述介绍,下面将基于子图像分割网络的训练方式,对本申请提供的模型训练方法进行介绍,请参阅图12,本申请实施例中模型训练的方法一个实施例包括:
[0239] 301、获取待训练图像,其中,待训练图像为已经过标注的图像;
[0240] 本实施例中,模型训练装置获取待训练图像,可以理解的是,待训练图像是已经经过标注的图像,且训练的过程中需要采用大量的待训练图像进行训练,为了便于说明,这里将以对一个待训练图像为例进行介绍。模型训练装置通常部署于服务器。
[0241] 待训练图像可以是包括四个模态的BraTS18比赛用MR数据集,四个模态分别为T1、T2、T1C和FLAIR。当然,在实际应用中还可以采用其他类型的待训练图像。接下来,需要对待训练图像进行归一化处理,对于没有固定的形状大小和方向性的病灶区域,可以对该待训练图像进行翻转、旋转、放缩和对比度增强等数据增广的操作,从而增加训练样本数量,增大方向性和不同大小规模下的信息价值。
[0242] 302、通过待训练子图像分割网络获取待训练图像所对应的预测图像;
[0243] 本实施例中,模型训练装置将待训练图像输入至待训练子图像分割网络,待训练子图像分割网络用于对图像进行分割,从而得到分割后的预测图像。
[0244] 303、采用待训练图像与预测图像对待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0245] 本实施例中,模型训练装置基于待训练图像和预测图像,采用损失函数进行计算。具体地,不同类型的待训练图像分割网可以采用不同的损失函数,也可以基于不同的分割模型类型,选择合适的损失函数。当损失函数收敛时,获取对应的网络模型参数。
[0246] 304、根据网络模型参数生成子图像分割网络,其中,子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,编码图像为自编码器对原始图像进行编码后得到的,分割结果用于确定原始图像的不确定度,不确定度用于确定待标注图像,原始图像为未经过标注的图像。
[0247] 本实施例中,模型训练装置根据生成的网络模型参数,将该网络模型参数作为子图像分割网络的模型参数,从而得到最终的子图像分割网络。于是,在预测的过程中,可以使用该子图像分割网络对原始图像进行分割。
[0248] 可以理解的是,在实际应用中,需要对多个子图像分割网络都执行如步骤301至步骤304的操作,这里仅以训练一个子图像分割网络为例进行介绍,然而这不应理解为对本申请的限定。
[0249] 本申请实施例中,提供了一种子图像分割网络的训练方法,即首先获取待训练图像,然后通过待训练子图像分割网络获取待训练图像所对应的预测图像,再采用待训练图像与预测图像对待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数,最后根据网络模型参数生成子图像分割网络。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0250] 下面对本申请中的待标注图像选择装置进行详细描述,请参阅图13,图13为本申请实施例中待标注图像选择装置一个实施例示意图,待标注图像选择装置40包括:
[0251] 获取模块401,用于获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
[0252] 所述获取模块401,还用于通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
[0253] 所述获取模块401,还用于通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
[0254] 确定模块402,用于根据所述获取模块401获取的所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
[0255] 本实施例中,获取模块401获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数,所述获取模块401通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系,所述获取模块401通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数,确定模块402根据所述获取模块401获取的所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
[0256] 本申请实施例中,提供了一种待标注图像确定装置,首先获取原始图像以及自编码器集合,原始图像为未经过标注的图像,然后通过自编码器集合获取原始图像所对应的编码图像集合,编码图像集合包括N个编码图像,编码图像与自编码器之间具有对应关系,再通过图像分割网络获取编码图像集合以及原始图像所对应的分割结果集合,图像分割网络包括M个子图像分割网络,分割结果集合包括[(N+1)*M]个分割结果,最后根据分割结果集合确定原始图像所对应的不确定度,其中,不确定度用于确定待标注图像。通过上述方式,将原始图像通过不同编码策略的自编码器进行改变,不同的自编码器直接能够侧重提取该原始图像不同方面的信息,分化出不同自编码器对原始图像的理解,使得一部分重建后相似而另一部分细节就出现差异。且不同的自编码器输出结果应用在不同的子图像分割网络上,使得输出的结果变化更多,从而提升不确定度的计算合理性,进而有助于选择更加准确的待标注图像。
[0257] 可选地,在上述图13所对应的实施例的基础上,本申请实施例提供的待标注图像选择装置40的另一实施例中,
[0258] 所述获取模块401,具体用于通过第一自编码器获取所述原始图像所对应的第一编码图像;
[0259] 通过第二自编码器获取所述原始图像所对应的第二编码图像,其中,所述第二自编码器与所述第一自编码器属于两种不同类型的自编码器,所述第二编码图像和所述第一编码图像属于两个不同的图像。
[0260] 其次,本申请实施例中,提供了一种基于多个自编码器生成编码图像的方法,即通过第一自编码器获取原始图像所对应的第一编码图像,并且通过第二自编码器获取原始图像所对应的第二编码图像,第二自编码器与第一自编码器属不同类型的自编码器。通过上述方式,能够利用多个自编码器对图像进行编码,由于不同自编码器能侧重提取图像的不同特征进行编码,利用多个自编码器生成的不同图像在主干位置变换不大而细节变化较大,由此,可以将网络理解的细节定义为“不确定度”,这部分就是自编码器前后改变较大的区域,从而分割时会造成不确定度大的区域分割结果差异也较大,提升不确定度计算的可靠性。
[0261] 可选地,在上述图13所对应的实施例的基础上,本申请实施例提供的待标注图像选择装置40的另一实施例中,
[0262] 所述获取模块401,具体用于通过第一子图像分割网络获取第一分割结果,其中,所述第一分割结果包括(N+1)个第一分割子结果,所述第一分割子结果与所述编码图像和所述原始图像具有对应关系;
[0263] 通过第二子图像分割网络获取第二分割结果,其中,所述第二分割结果包括(N+1)个第二分割子结果,所述第二分割子结果与所述编码图像和所述原始图像具有对应关系,所述第二子图像分割网络与所述第一子图像分割网络属于两种不同类型的子图像分割网络。
[0264] 其次,本申请实施例中,提供了一种基于多个子图像分割网络生成分割结果的方法,即通过第一子图像分割网络获取第一分割结果,通过第二子图像分割网络获取第二分割结果,其中,第二子图像分割网络与第一子图像分割网络属于不同类型的子图像分割网络。通过上述方式,能够利用多个子图像分割网络对图像进行分割,由于不同子图像分割网络能侧重提取图像的分割区域,采用不同的自编码器输出结果应用在不同的子图像分割网络上,从而使输出图像变化更多,不确定度的结果也更加精确。
[0265] 可选地,在上述图13所对应的实施例的基础上,本申请实施例提供的待标注图像选择装置40的另一实施例中,
[0266] 所述确定模块402,具体用于获取所述原始图像中每个像素点所对应的灰度值;
[0267] 根据所述分割结果集合以及所述原始图像中每个像素点所对应的灰度值,确定所述原始图像中每个像素点所对应的熵集合,其中,所述熵集合包括[(N+1)*M]个熵;
[0268] 根据所述原始图像中每个像素点所对应的熵集合,计算得到所述原始图像所对应的不确定度。
[0269] 再次,本申请实施例中,提供了一种计算不确定度的方式,即获取原始图像中每个像素点所对应的灰度值,然后根据分割结果集合以及原始图像中每个像素点所对应的灰度值,确定原始图像中每个像素点所对应的熵集合,最后根据原始图像中每个像素点所对应的熵集合,计算得到原始图像所对应的不确定度。通过上述方式,为不确定度的计算提供了可行的实现方式,从而提升方案的可操作性和可靠性。
[0270] 可选地,在上述图13所对应的实施例的基础上,本申请实施例提供的待标注图像选择装置40的另一实施例中,
[0271] 所述确定模块402,具体用于获取每个分割结果中每个像素点所对应的灰度值;
[0272] 基于所述分割结果集合,根据所述每个分割结果中每个像素点所对应的灰度值,确定每个像素点的方差;
[0273] 根据所述每个像素点的方差以及像素点总数,确定所述原始图像所对应的不确定度。
[0274] 再次,本申请实施例中,提供了另一种计算不确定度的方式,即首先获取每个分割结果中每个像素点所对应的灰度值,然后基于分割结果集合,根据每个分割结果中每个像素点所对应的灰度值,确定每个像素点的方差,最后根据每个像素点的方差以及像素点总数,确定原始图像所对应的不确定度。通过上述方式,为不确定度的计算提供了另一种可行的实现方式,从而提升方案的可操作性、可靠性以及灵活性。
[0275] 可选地,在上述图13所对应的实施例的基础上,请参阅图14,本申请实施例提供的待标注图像选择装置40的另一实施例中,所述待标注图像确定装置40还包括判断模块403;
[0276] 所述判断模块403,用于在所述确定模块402根据所述分割结果集合确定所述原始图像所对应的不确定度之后,判断所述不确定度是否大于或等于不确定度门限;
[0277] 所述确定模块402,还用于若所述判断模块403判断得到所述不确定度大于或等于所述不确定度门限,则将所述原始图像确定为所述待标注图像。
[0278] 其次,本申请实施例中,提供了一种基于不确定度自行提供待标注图像的方法,在根据分割结果集合确定原始图像所对应的不确定度之后,还可以判断不确定度是否大于或等于不确定度门限,若不确定度大于或等于不确定度门限,则将原始图像确定为待标注图像。通过上述方式,能够根据不确定度自动判断该原始图像是否适合作为待标注图像,这样的话,一方面能够提升待标注图像选择的效率,无需人为根据不确定度进行待标注图像的筛选,另一方面,能够有效地提升待标注图像的选择的准确度,防止出现待标注图像漏选的情况。
[0279] 可选地,在上述图13所对应的实施例的基础上,请参阅图15,本申请实施例提供的待标注图像选择装置40的另一实施例中,所述待标注图像确定装置还包括编码模块404、解码模块405、训练模块406以及生成模块407;
[0280] 所述获取模块401,还用于获取原始图像以及自编码器集合之前,获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0281] 所述编码模块404,用于通过待训练自编码器的编码器对所述获取模块401获取的所述待训练图像进行编码,得到编码结果;
[0282] 所述解码模块405,用于通过所述待训练自编码器的解码器,对所述编码模块404编码得到的所述编码结果进行解码,得到预测图像;
[0283] 所述训练模块406,用于采用所述待训练图像与所述解码模块405解码得到的所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0284] 所述生成模块407,用于根据所述训练模块406训练得到的所述自编码器模型参数生成自编码器,其中,所述自编码器为所述自编码器集合中的一个自编码器。
[0285] 其次,本申请实施例中,提供了一种自编码器的训练方法,即首先获取待训练图像,然后通过待训练自编码器的编码器对待训练图像进行编码,得到编码结果,再通过待训练自编码器的解码器对编码结果进行解码,得到预测图像在,最后采用待训练图像与预测图像对待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数,利用自编码器模型参数生成自编码器。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0286] 可选地,在上述图13所对应的实施例的基础上,请参阅图16,本申请实施例提供的待标注图像选择装置40的另一实施例中,所述待标注图像确定装置40还包括训练模块406以及生成模块407;
[0287] 所述获取模块401,还用于通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合之前,获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0288] 所述获取模块401,还用于通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0289] 所述训练模块406,用于采用所述待训练图像与所述获取模块401获取的所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0290] 所述生成模块407,用于根据所述训练模块406训练得到的所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络为所述图像分割网络中的一个子图像分割网络。
[0291] 其次,本申请实施例中,提供了一种子图像分割网络的训练方法,即首先获取待训练图像,然后通过待训练子图像分割网络获取待训练图像所对应的预测图像,再采用待训练图像与预测图像对待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数,最后根据网络模型参数生成子图像分割网络。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0292] 下面对本申请中的模型训练装置进行详细描述,请参阅图17,图17为本申请实施例中模型训练装置一个实施例示意图,模型训练装置50包括:
[0293] 获取模块501,用于获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0294] 编码模块502,用于通过待训练自编码器的编码器对所述获取模块501获取的所述待训练图像进行编码,得到编码结果;
[0295] 解码模块503,用于通过所述待训练自编码器的解码器,对所述编码模块502编码后的所述编码结果进行解码,得到预测图像;
[0296] 训练模块504,用于采用所述待训练图像与所述解码模块503解码得到的所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0297] 生成模块505,用于根据所述训练模块504训练得到的所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0298] 本实施例中,获取模块501获取待训练图像,其中,所述待训练图像为已经过标注的图像,编码模块502通过待训练自编码器的编码器对所述获取模块501获取的所述待训练图像进行编码,得到编码结果,解码模块503通过所述待训练自编码器的解码器,对所述编码模块502编码后的所述编码结果进行解码,得到预测图像,训练模块504采用所述待训练图像与所述解码模块503解码得到的所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数,生成模块505根据所述训练模块504训练得到的所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0299] 本申请实施例中,提供了一种模型训练装置,即首先获取待训练图像,然后通过待训练自编码器的编码器对待训练图像进行编码,得到编码结果,再通过待训练自编码器的解码器对编码结果进行解码,得到预测图像在,最后采用待训练图像与预测图像对待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数,利用自编码器模型参数生成自编码器。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0300] 下面对本申请中的模型训练装置进行详细描述,请参阅图18,图18为本申请实施例中模型训练装置一个实施例示意图,模型训练装置60包括:
[0301] 获取模块601,用于获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0302] 所述获取模块601,还用于通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0303] 训练模块602,用于采用所述待训练图像与所述获取模块601获取的所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0304] 生成模块603,用于根据所述训练模块602训练得到的所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0305] 本实施例中,获取模块601获取待训练图像,其中,所述待训练图像为已经过标注的图像,所述获取模块601通过待训练子图像分割网络获取所述待训练图像所对应的预测图像,训练模块602采用所述待训练图像与所述获取模块601获取的所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数,生成模块603根据所述训练模块602训练得到的所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0306] 本申请实施例中,提供了一种模型训练装置,即首先获取待训练图像,然后通过待训练子图像分割网络获取待训练图像所对应的预测图像,再采用待训练图像与预测图像对待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数,最后根据网络模型参数生成子图像分割网络。通过上述方式,为自编码器的训练提供了可行的实现方式,从而提升方案的可行性和可操作性。
[0307] 图19是本申请实施例提供的一种服务器结构示意图,该服务器700可因配置或性能不同而产生比较大的差异,可以包括一个或一个以上中央处理器(central processing units,CPU)722(例如,一个或一个以上处理器)和存储器732,一个或一个以上存储应用程序742或数据744的存储介质730(例如一个或一个以上海量存储设备)。其中,存储器732和存储介质730可以是短暂存储或持久存储。存储在存储介质730的程序可以包括一个或一个以上模块(图示没标出),每个模块可以包括对服务器中的一系列指令操作。更进一步地,中央处理器722可以设置为与存储介质730通信,在服务器700上执行存储介质730中的一系列指令操作。
[0308] 服务器700还可以包括一个或一个以上电源726,一个或一个以上有线或无线网络接口750,一个或一个以上输入输出接口758,和/或,一个或一个以上操作系统741,例如Windows ServerTM,Mac OS XTM,UnixTM,LinuxTM,FreeBSDTM等等。
[0309] 上述实施例中由服务器所执行的步骤可以基于该图19所示的服务器结构。
[0310] 在本申请实施例中,该终端所包括的处理器880还具有以下功能:
[0311] 获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
[0312] 通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
[0313] 通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
[0314] 根据所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
[0315] 在本申请实施例中,该终端所包括的处理器880还具有以下功能:
[0316] 获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0317] 通过待训练自编码器的编码器对所述待训练图像进行编码,得到编码结果;
[0318] 通过所述待训练自编码器的解码器对所述编码结果进行解码,得到预测图像;
[0319] 采用所述待训练图像与所述预测图像对所述待训练自编码器进行训练,直至损失函数收敛,获取自编码器模型参数;
[0320] 根据所述自编码器模型参数生成自编码器,其中,所述自编码器用于对原始图像进行编码,得到编码图像,所述编码图像用于通过子图像分割网络生成分割结果,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0321] 在本申请实施例中,该终端所包括的处理器880还具有以下功能:
[0322] 获取待训练图像,其中,所述待训练图像为已经过标注的图像;
[0323] 通过待训练子图像分割网络获取所述待训练图像所对应的预测图像;
[0324] 采用所述待训练图像与所述预测图像对所述待训练子图像分割网络进行训练,直至损失函数收敛,获取网络模型参数;
[0325] 根据所述网络模型参数生成子图像分割网络,其中,所述子图像分割网络用于对编码图像以及原始图像进行处理,得到多个分割结果,所述编码图像为自编码器对所述原始图像进行编码后得到的,所述分割结果用于确定所述原始图像的不确定度,所述不确定度用于确定待标注图像,所述原始图像为未经过标注的图像。
[0326] 本申请实施例还提供了另一种待标注图像选择装置,如图20所示,为了便于说明,仅示出了与本申请实施例相关的部分,具体技术细节未揭示的,请参照本申请实施例方法部分。该终端设备可以为包括手机、平板电脑、个人数字助理(Personal  Digital Assistant,PDA)、销售终端设备(Point of Sales,POS)、车载电脑等任意终端设备设备,以终端设备为手机为例:
[0327] 图20示出的是与本申请实施例提供的终端设备相关的手机的部分结构的框图。参考图20,手机包括:射频(Radio Frequency,RF)电路810、存储器820、输入单元830、显示单元840、传感器850、音频电路860、无线保真(wireless fidelity,WiFi)模块870、处理器880、以及电源890等部件。本领域技术人员可以理解,图20中示出的手机结构并不构成对手机的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件布置。
[0328] 下面结合图20对手机的各个构成部件进行具体的介绍:
[0329] RF电路810可用于收发信息或通话过程中,信号的接收和发送,特别地,将基站的下行信息接收后,给处理器880处理;另外,将设计上行的数据发送给基站。通常,RF电路810包括但不限于天线、至少一个放大器收发信机耦合器低噪声放大器(Low Noise Amplifier,LNA)、双工器等。此外,RF电路810还可以通过无线通信与网络和其他设备通信。上述无线通信可以使用任一通信标准或协议,包括但不限于全球移动通讯系统(Global System of Mobile communication,GSM)、通用分组无线服务(General Packet Radio Service,GPRS)、码分多址(Code Division Multiple Access,CDMA)、宽带码分多址(Wideband Code Division Multiple Access,WCDMA)、长期演进(Long Term Evolution,LTE)、电子邮件、短消息服务(Short Messaging Service,SMS)等。
[0330] 存储器820可用于存储软件程序以及模块,处理器880通过运行存储在存储器820的软件程序以及模块,从而执行手机的各种功能应用以及数据处理。存储器820可主要包括存储程序区和存储数据区,其中,存储程序区可存储操作系统、至少一个功能所需的应用程序(比如声音播放功能、图像播放功能等)等;存储数据区可存储根据手机的使用所创建的数据(比如音频数据、电话本等)等。此外,存储器820可以包括高速随机存取存储器,还可以包括非易失性存储器,例如至少一个磁盘存储器件、闪存器件、或其他易失性固态存储器件。
[0331] 输入单元830可用于接收输入的数字或字符信息,以及产生与手机的用户设置以及功能控制有关的键信号输入。具体地,输入单元830可包括触控面板831以及其他输入设备832。触控面板831,也称为触摸屏,可收集用户在其上或附近的触摸操作(比如用户使用手指、触笔等任何适合的物体或附件在触控面板831上或在触控面板831附近的操作),并根据预先设定的程式驱动相应的连接装置。可选的,触控面板831可包括触摸检测装置和触摸控制器两个部分。其中,触摸检测装置检测用户的触摸方位,并检测触摸操作带来的信号,将信号传送给触摸控制器;触摸控制器从触摸检测装置上接收触摸信息,并将它转换成触点坐标,再送给处理器880,并能接收处理器880发来的命令并加以执行。此外,可以采用电阻式、电容式、红外线以及表面声波等多种类型实现触控面板831。除了触控面板831,输入单元830还可以包括其他输入设备832。具体地,其他输入设备832可以包括但不限于物理键盘、功能键(比如音量控制按键、开关按键等)、轨迹球鼠标、操作杆等中的一种或多种。
[0332] 显示单元840可用于显示由用户输入的信息或提供给用户的信息以及手机的各种菜单。显示单元840可包括显示面板841,可选的,可以采用液晶显示器(Liquid Crystal Display,LCD)、有机发光二极管(Organic Light-Emitting Diode,OLED)等形式来配置显示面板841。进一步的,触控面板831可覆盖显示面板841,当触控面板831检测到在其上或附近的触摸操作后,传送给处理器880以确定触摸事件的类型,随后处理器880根据触摸事件的类型在显示面板841上提供相应的视觉输出。虽然在图20中,触控面板831与显示面板841是作为两个独立的部件来实现手机的输入和输入功能,但是在某些实施例中,可以将触控面板831与显示面板841集成而实现手机的输入和输出功能。
[0333] 手机还可包括至少一种传感器850,比如光传感器、运动传感器以及其他传感器。具体地,光传感器可包括环境光传感器及接近传感器,其中,环境光传感器可根据环境光线的明暗来调节显示面板841的亮度,接近传感器可在手机移动到边时,关闭显示面板841和/或背光。作为运动传感器的一种,加速计传感器可检测各个方向上(一般为三轴)加速度的大小,静止时可检测出重力的大小及方向,可用于识别手机姿态的应用(比如横竖屏切换、相关游戏、磁力计姿态校准)、振动识别相关功能(比如计步器、敲击)等;至于手机还可配置的陀螺仪、气压计、湿度计、温度计、红外线传感器等其他传感器,在此不再赘述。
[0334] 音频电路860、扬声器861,传声器862可提供用户与手机之间的音频接口。音频电路860可将接收到的音频数据转换后的电信号,传输到扬声器861,由扬声器861转换为声音信号输出;另一方面,传声器862将收集的声音信号转换为电信号,由音频电路860接收后转换为音频数据,再将音频数据输出处理器880处理后,经RF电路810以发送给比如另一手机,或者将音频数据输出至存储器820以便进一步处理。
[0335] WiFi属于短距离无线传输技术,手机通过WiFi模块870可以帮助用户收发电子邮件、浏览网页和访问流式媒体等,它为用户提供了无线的宽带互联网访问。虽然图20示出了WiFi模块870,但是可以理解的是,其并不属于手机的必须构成,完全可以根据需要在不改变发明的本质的范围内而省略。
[0336] 处理器880是手机的控制中心,利用各种接口和线路连接整个手机的各个部分,通过运行或执行存储在存储器820内的软件程序和/或模块,以及调用存储在存储器820内的数据,执行手机的各种功能和处理数据,从而对手机进行整体监控。可选的,处理器880可包括一个或多个处理单元;可选的,处理器880可集成应用处理器和调制解调处理器,其中,应用处理器主要处理操作系统、用户界面和应用程序等,调制解调处理器主要处理无线通信。可以理解的是,上述调制解调处理器也可以不集成到处理器880中。
[0337] 手机还包括给各个部件供电的电源890(比如电池),可选的,电源可以通过电源管理系统与处理器880逻辑相连,从而通过电源管理系统实现管理充电、放电、以及功耗管理等功能。
[0338] 尽管未示出,手机还可以包括摄像头、蓝牙模块等,在此不再赘述。
[0339] 在本申请实施例中,该终端设备所包括的处理器880还具有以下功能:
[0340] 获取原始图像以及自编码器集合,其中,所述原始图像为未经过标注的图像,所述自编码器集合包括N个自编码器,所述N为大于或大于1的整数;
[0341] 通过自编码器集合获取所述原始图像所对应的编码图像集合,其中,所述编码图像集合包括N个编码图像,所述编码图像与所述自编码器之间具有对应关系;
[0342] 通过图像分割网络获取所述编码图像集合以及所述原始图像所对应的分割结果集合,其中,所述图像分割网络包括M个子图像分割网络,所述分割结果集合包括[(N+1)*M]个分割结果,所述M为大于或等于1的整数;
[0343] 根据所述分割结果集合确定所述原始图像所对应的不确定度,其中,所述不确定度用于确定待标注图像。
[0344] 所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的系统,装置和单元的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
[0345] 在本申请所提供的几个实施例中,应该理解到,所揭露的系统,装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
[0346] 所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
[0347] 另外,在本申请各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。
[0348] 所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的全部或部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本申请各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘只读存储器(read-only memory,ROM)、随机存取存储器(random access memory,RAM)、磁碟或者光盘等各种可以存储程序代码的介质。
[0349] 以上所述,以上实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围。
高效检索全球专利

专利汇是专利免费检索,专利查询,专利分析-国家发明专利查询检索分析平台,是提供专利分析,专利查询,专利检索等数据服务功能的知识产权数据服务商。

我们的产品包含105个国家的1.26亿组数据,免费查、免费专利分析。

申请试用

分析报告

专利汇分析报告产品可以对行业情报数据进行梳理分析,涉及维度包括行业专利基本状况分析、地域分析、技术分析、发明人分析、申请人分析、专利权人分析、失效分析、核心专利分析、法律分析、研发重点分析、企业专利处境分析、技术处境分析、专利寿命分析、企业定位分析、引证分析等超过60个分析角度,系统通过AI智能系统对图表进行解读,只需1分钟,一键生成行业专利分析报告。

申请试用

QQ群二维码
意见反馈