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

一种舌象分割方法、设备及存储介质与流程

2021-10-29 21:02:00 来源:中国专利 TAG:分割 图像 方法 设备 存储介质


1.本技术涉及图像分割技术领域,尤其是涉及一种舌象分割方法、设备及存储介质。


背景技术:

2.中医舌诊是中医望闻问切中的第一步,也是最直观的一步,舌象能够把人体内的生理状况和病理变化一一展现给医师看,为医师诊断起到重要的作用。因此,舌诊的客观化和定量化也成为我国中医现代化研究的重要课题。
3.随着图像处理技术在各个领域得到越来越多的应用,舌象分割技术也越来越多,根据有无标签,分为四种网络结构:全监督分割网络结构、半监督分割网络结构、弱监督分割网络结构、无监督分割网络结构。其中全监督分割分割精度最高,但需要对每一个像素进行标注,需要花费大量的人力、物力;半监督和弱监督虽然分割进度相对较慢,但不需要对每一个像素做标签,能够节省大量的时间。
4.现有的舌象分割技术大部分都是基于传统技术分割方法,虽然具备训练速度快等优点,但需要根据实验手动选择特征,自动化低,识别率也不高。因此有研究开始向深度学习图像分割技术方面研究舌象分割,但这些研究都是建立在全监督分割网络结构上进行研究,需要大量的数据集与大量的时间进行像素级标注,因此,本发明人认为现有的舌象分割技术还有进一步的改进空间。


技术实现要素:

5.有鉴于此,本技术提供一种舌象分割方法、设备及存储介质,用以解决现有基于深度学习的舌象分割技术中需要花费大量人力、物力进行像素级标注的问题。
6.为了解决上述问题,第一方面,本发明提供一种舌象分割方法,所述方法包括:
7.获取舌象训练集,所述舌象训练集包括多张人体舌象图像;
8.对所述舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;
9.将所述舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新所述条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;
10.获取待识别舌象图像,并输入至所述训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
11.可选的,获取舌象训练集包括:
12.获取原始舌象数据集;
13.对所述原始舌象数据集进行图像预处理后,得到舌象训练集,其中所述图像预处理包括图像旋转、平移交换、数据增强以及对比度归一化处理。
14.可选的,对所述舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集,包括:
15.将所述舌象训练集的每张人体舌象图像输入至预设的舌象特征提取网络,其中所述舌象特征提取网络采用训练完成的vgg16网络;
16.提取每张人体舌象图像经过两个预设的非连续卷积层后的舌体特征图,并将两个舌体特征图的维度和参数量处理为一致;
17.将两个舌体特征图进行融合后,得到舌体掩膜;由所述舌象训练集得到的舌体掩膜集作为真实标签集。
18.可选的,将两个舌体特征图进行融合后,得到舌体掩膜包括:
19.将两个舌体特征图从3d张量转换为2d矩阵,将两个结果矩阵进行融合,并将融合结果值在[0,1]区间进行缩放,得到目标矩阵,所述目标矩阵中每个元素的值作为对应像素的前景概率;由所述目标矩阵确定的图像确定为舌体掩膜;
[0020]
由所述舌象训练集得到的舌体掩膜集作为真实标签集之前,所述方法还包括:
[0021]
利用条件随机场对所述舌体掩膜进行平滑处理。
[0022]
可选的,所述条件生成对抗网络包括生成器、判别器以及预设的对抗损失函数;
[0023]
所述生成器由u

net网络和预设的生成损失函数构成;
[0024]
所述判别器由多个卷积层和一个判别层,以及预设的判别损失函数构成,其中所述多个卷积层的通道数设置随层级数依次增大。
[0025]
可选的,将所述舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代训练,分别更新所述条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络,包括:
[0026]
初始化迭代次数,以及初始化生成器和判别器的权值和偏置;
[0027]
开始迭代训练,包括:
[0028]
固定当前判别器的网络参数,对抗训练当前生成器,得到目标生成器;
[0029]
固定所述目标生成器的网络参数,对抗训练当前判别器,得到目标判别器;
[0030]
当迭代次数到达预设次数后,得到训练完成的条件生成对抗网络。
[0031]
可选的,所述固定当前判别器的网络参数,对抗训练当前生成器,得到目标生成器,包括:
[0032]
将舌象训练集中的人体舌象图像和成对的真实标签进行拼接后,得到真实舌象图像并输入至当前生成器,计算所述当前生成器的生成损失值,基于生成损失值反向传播训练所述当前生成器,以输出第一预测标签;
[0033]
将第一预测标签和对应的所述舌象训练集中的人体舌象图像进行拼接后,得到第一预测舌象图像;
[0034]
将所述第一预测舌象图像和所述真实舌象图像输入至所述当前判别器进行预测,得到两个不同的第一预测结果,分别是与所述第一预测舌象图像对应的第一预测结果、与所述真实舌象图像对应的第一预测结果;
[0035]
根据两个不同的所述第一预测结果,计算所述条件生成对抗网络的第一对抗损失值,采用梯度下降法并根据第一对抗损失值,对所述当前生成器的网络参数进行更新,以得到目标生成器。
[0036]
可选的,所述固定所述目标生成器的网络参数,对抗训练当前判别器,得到目标判别器,包括:
[0037]
将所述真实舌象图像输入至目标生成器,得到第二预测标签;
[0038]
将所述第二预测标签和对应的所述舌象训练集中的人体舌象图像进行拼接后,得到第二预测舌象图像;
[0039]
将所述第二预测舌象图像和所述真实舌象图像输入所述当前判别器,计算所述当前判别器的判别损失值,基于所述判别损失值反向传播训练当前判别器,以输出两个不同的第二预测结果,分别是与所述第二预测舌象图像对应的第二预测结果、与所述真实舌象图像对应的第二预测结果;
[0040]
根据两个不同的所述第二预测结果,计算所述条件生成对抗网络的第二对抗损失值,采用梯度下降法并根据所述第二对抗损失值,对所述当前判别器的网络参数进行更新后,得到目标判别器。
[0041]
第二方面,本发明提供一种舌象分割系统,所述系统包括:
[0042]
数据获取模块,用于获取舌象训练集,舌象训练集包括多张人体舌象图像;
[0043]
特征提取模块,用于对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;
[0044]
训练模块,用于将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;
[0045]
舌象分割模块,用于获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0046]
第三方面,本发明提供一种计算机设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如所述舌象分割方法的步骤。
[0047]
第四方面,本发明提供一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现如所述舌象分割方法的步骤。
[0048]
采用上述实施例的有益效果是:通过提取舌象训练集的特征信息即舌体掩膜,并作为网络训练的真实标签,便于引导条件生成对抗网络进行舌象分割,无需人工做标签,并且在舌象数据集小的情况下,也能够有效提高舌象分割精度。
附图说明
[0049]
图1为本发明提供的舌象分割系统的应用场景示意图;
[0050]
图2为本发明提供的舌象分割方法一实施例的方法流程图;
[0051]
图3为本发明提供的舌象分割方法步骤s201一实施例的方法流程图;
[0052]
图4为本发明提供的舌象分割方法步骤s202一实施例的方法流程图;
[0053]
图5为本发明提供的舌象分割方法步骤s403一实施例的方法流程图;
[0054]
图6为本发明提供的舌象分割方法步骤s203一实施例的方法流程图;
[0055]
图7为本发明提供的舌象分割方法步骤s6021一实施例的方法流程图;
[0056]
图8为本发明提供的舌象分割方法步骤s6022一实施例的方法流程图;
[0057]
图9为本发明提供的u

net网络和基于u

net的条件生成对抗网络对人体舌象图像分别分割后的示意图;
[0058]
图10为本发明提供的舌象分割系统一实施例的原理框图;
[0059]
图11为本发明提供的计算机设备一实施例的原理框图。
具体实施方式
[0060]
下面结合附图来具体描述本发明的优选实施例,其中,附图构成本技术一部分,并与本发明的实施例一起用于阐释本发明的原理,并非用于限定本发明的范围。
[0061]
在本技术的描述中,“多个”的含义是两个或两个以上,除非另有明确具体的限定。
[0062]
在本文中提及“实施例”意味着,结合实施例描述的特定特征、结构或特性可以包含在本发明的至少一个实施例中。在说明书中的各个位置出现该短语并不一定均是指相同的实施例,也不是与其它实施例互斥的独立的或备选的实施例。本领域技术人员显式地和隐式地理解的是,本文所描述的实施例可以与其它实施例相结合。
[0063]
本发明提供了一种舌象分割方法、系统、设备及存储介质,以下分别进行详细说明。
[0064]
图1为本技术实施例所提供的舌象分割系统的场景示意图,该系统可以包括服务器100,服务器100中集成有舌象分割系统,如图1中的服务器。
[0065]
本技术实施例中服务器100主要用于:
[0066]
获取舌象训练集,舌象训练集包括多张人体舌象图像;对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0067]
本技术实施例中,该服务器100可以是独立的服务器,也可以是服务器组成的服务器网络或服务器集群,例如,本技术实施例中所描述的服务器100,其包括但不限于计算机、网络主机、单个网络服务器、多个网络服务器集或多个服务器构成的云服务器。其中,云服务器由基于云计算(cloud computing)的大量计算机或网络服务器构成。
[0068]
可以理解的是,本技术实施例中所使用的终端200可以是既包括接收和发射硬件的设备,即具有能够在双向通信链路上,执行双向通信的接收和发射硬件的设备。这种设备可以包括:蜂窝或其他通信设备,其具有单线路显示器或多线路显示器或没有多线路显示器的蜂窝或其他通信设备。具体的终端200可以是台式机、便携式电脑、网络服务器、掌上电脑(personal digital assistant,pda)、移动手机、平板电脑、无线终端设备、通信设备、嵌入式设备等,本实施例不限定终端200的类型。
[0069]
本领域技术人员可以理解,图1中示出的应用环境,仅仅是与本技术方案一种应用场景,并不构成对本技术方案应用场景的限定,其他的应用环境还可以包括比图1中所示更多或更少的终端,例如图1中仅示出2个终端,可以理解的,该舌象分割系统还可以包括一个或多个其他终端,具体此处不作限定。
[0070]
另外,参照图1所示,该舌象分割系统还可以包括存储器200,用于存储数据,如舌象训练集、真实标签集等。
[0071]
需要说明的是,图1所示的舌象分割系统的场景示意图仅仅是一个示例,本技术实施例描述的舌象分割系统以及场景是为了更加清楚的说明本技术实施例的技术方案,并不
构成对于本技术实施例提供的技术方案的限定,本领域普通技术人员可知,随着舌象分割系统的演变和新业务场景的出现,本技术实施例提供的技术方案对于类似的技术问题,同样适用。
[0072]
首先,本技术实施例提供了一种舌象分割方法,该方法包括:获取舌象训练集,舌象训练集包括多张人体舌象图像;对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0073]
参照图2,本技术提供的舌象分割方法一种实施例的方法流程图,该舌象分割方法包括下述步骤:
[0074]
s201、获取舌象训练集,舌象训练集包括多张人体舌象图像;
[0075]
s202、对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;
[0076]
s203、将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;
[0077]
s204、获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0078]
本实施例的舌象分割方法通过提取舌象训练集的特征信息即舌体掩膜,并作为网络训练的真实标签,便于引导条件生成对抗网络进行舌象分割,无需人工做标签,并且在舌象数据集小的情况下,也能够有效提供舌象分割精度。
[0079]
需要说明的是,对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集,该舌体掩膜集是指感兴趣区掩模集合,感兴趣区是人的舌体。将舌体掩膜集作为真实标签,并作为一种先验条件,将舌体掩膜集与舌象数据集一起输入条件生成对抗网络,便于对抗训练。
[0080]
参照图3,本技术提供的步骤s201一种实施例的方法流程图,该步骤s201包括下述步骤:
[0081]
s301、获取原始舌象数据集;
[0082]
s302、对原始舌象数据集进行图像预处理后,得到舌象训练集,其中图像预处理包括图像旋转、平移交换、数据增强以及对比度归一化处理。
[0083]
在一实施例中,原始舌象数据集可以采用互联网上公开的在线舌象数据集tonguedatasets,该数据集包含为300张人体舌象图片。对原始舌象数据集进行图像预处理,具体地,采用图像旋转,平移交换,对原始舌象数据进行扩张增强,并通过对比度增强,对图像进行归一化处理,从而降低训练数据的拟合度。
[0084]
参照图4,本技术提供的步骤s202一种实施例的方法流程图,该步骤s202包括下述步骤:
[0085]
s401、将舌象训练集的每张人体舌象图像输入至预设的舌象特征提取网络,其中舌象特征提取网络采用训练完成的vgg16网络;
[0086]
s402、提取每张人体舌象图像经过两个预设的非连续卷积层后的舌体特征图,并将两个舌体特征图的维度和参数量处理为一致;
[0087]
s403、将两个舌体特征图进行融合后,得到舌体掩膜;由舌象训练集得到的舌体掩膜集作为真实标签集。
[0088]
在一实施例中,vgg16网络总共有16层,13个卷积层和3个全连接层,具体地,将某张图像输入至vgg16网络,第一步经过两次[3*3]的64通道的卷积,获得一个[512,512,64]的初步有效特征层,再经过[2*2]的最大池化,得到一个[256,256,64]的特征层;第二步经过两次[3*3]的128通道的卷积,获得一个[256,256,128]的初步有效特征层,再经过[2*2]的最大池化,得到一个[128,128,128]的特征层;第三步经过三次[3*3]的256通道的卷积,获得一个[128,128,256]的初步有效特征层,再经过2*2最大池化,得到一个[64,64,256]的特征层;第四步经过三次[3*3]的512通道的卷积,获得一个[64,64,512]的初步有效特征层,再经过[2*2]最大池化,得到一个[32,32,512]的特征层;第五步经过三次[3*3]的512通道的卷积,获得一个[32,32,512]的初步有效特征层。需要说明的是,可以采用反向传播算法预先将vgg16训练好。
[0089]
在一实施例中,提取每张人体舌象图像经过两个预设的非连续卷积层后的舌体特征图,具体地,将舌象图像经过每一个卷积层提取特征后,都将对应的特征图可视化显示,通过对比分析,在图像经过第四层后卷积提取出来的特征图与舌象相吻合;进一步地,将第七层卷积后的特征图提取出来,并进行上采样操作,以恢复与第四层提取的特征图具有一样的数据维度和参数量,进一步地,将两张特征图融合成一个掩膜,亦即舌体掩膜。
[0090]
参照图5,本技术提供的步骤s403一种实施例的方法流程图,该步骤s403包括下述步骤:
[0091]
s501、将两个舌体特征图从3d张量转换为2d矩阵,将两个结果矩阵进行融合,并将融合结果值在[0,1]区间进行缩放,得到目标矩阵,目标矩阵中每个元素的值作为对应像素的前景概率;由目标矩阵确定的图像确定为舌体掩膜。
[0092]
s502、利用条件随机场对舌体掩膜进行平滑处理。
[0093]
在一实施例中,通过512卷积通道的平均池化操作将3d张量(512*w*h)转换为2d矩阵(w*h),然后通过元素求和将两个舌体特征图的2d结果矩阵进行融合,并将融合结果值缩放到0和1之间,以此产生的目标矩阵(融合图)为像素级的前景概率分布,该融合图即为舌体掩膜。
[0094]
考虑到步骤s501得到的舌体掩膜仍存在噪声,将舌体掩膜的前景概率分布输入条件随机场,对舌体掩膜进行平滑去噪,从而有助于提高后续条件生成对抗网络分割的准确率。
[0095]
在本实施例中,条件生成对抗网络包括生成器、判别器以及预设的对抗损失函数;生成器由u

net网络和预设的生成损失函数构成;判别器由多个卷积层和一个判别层,以及预设的判别损失函数构成,其中多个卷积层的通道数设置随层级数依次增大。
[0096]
在一实施例中,u

net网络由三部分组成,第一部分为主干特征提取部分,为卷积层和最大池化的堆叠,利用主干提取特征层可以获得五个初步的有效特征层。在第二部分中,利用这五个有效特征层进行特征融合,具体地,利用主干部分提取获得到的五个初步有效特征层进行上采样,并且进行特征融合,获得一个最终的,融合所有特征的有效特征层,
起到加强特征作用。第三部分是预测部分,利用第二部分最终获得的有效特征层对每一个特征点进行分类。
[0097]
生成器的生成损失函数具体如下:
[0098][0099]
式中,lg表示生成损失值;x
i
表示舌象训练集中第i个舌象图像,y
i
表示x
i
的真实标签,g(x
i
)表示生成器对舌象图像x
i
的预测标签;λ(g(x
i
),y
i
)表示每一对生成器对舌象图像x
i
的预测标签g(x
i
)和真实标签y
i
的损失函数。
[0100]
在一实施例中,判断器采用四层卷积层和一层判别层,第一层到第四层的通道数依次为64、128、256、512,卷积核大小均为4*4,步长为2;判别层的通道数为1,卷积核为1*1,步长为1。
[0101]
判别器的判别损失函数具体如下:
[0102][0103]
式中,d(x
i
,y
i
)表示判别器对带真实标签y
i
的舌象图像x
i
的预测结果;d(x
i
,g(x
i
))表示判别器对带预测标签g(x
i
)的舌象图像x
i
的预测结果;λ(d(x
i
,y
i
),1)表示每一对判别器对带真实标签y
i
的舌象图像x
i
和生成器生成标签1的损失函数;λ(d(x
i
,g(x
i
)),0)表示每一对判别器对带预测标签g(x
i
)的舌象图像x
i
和生成器生成标签0的损失函数。
[0104]
条件生成对抗网络的损失函数具体如下:
[0105]
l
cagn
=ex
i,
y
i
[log d(x
i
,y
i
)] ex
i
[log(1

d(x
i
,g(x
i
)))] λex
i
,y
i
[||y
i

g(x
i
)||1]
[0106]
式中,ex
i
,y
i
表示判别器对带真实标签y
i
的舌象图像x
i
的输出期望值;ex
i
表示判别器对带预测标签g(x
i
)的舌象图像x
i
的反向输出期望值;||y
i

g(x
i
)||1表示真实标签y
i
减去预测标签g(x
i
)的结果的l1范数;λ表示损失权重系数;l1范数是指向量中各个元素绝对值之和。
[0107]
参照图6,本技术提供的步骤s203一种实施例的方法流程图,该步骤s203包括下述步骤:
[0108]
s601、初始化迭代次数,以及初始化生成器和判别器的权值和偏置;
[0109]
s602、开始迭代训练,包括:
[0110]
s6021、固定当前判别器的网络参数,对抗训练当前生成器,得到目标生成器;
[0111]
s6022、固定目标生成器的网络参数,对抗训练当前判别器,得到目标判别器;
[0112]
s6023、当迭代次数到达预设次数后,得到训练完成的条件生成对抗网络。
[0113]
在一实施例中,设迭代次数为q,最大迭代次数为q,初始化令q=0。当q=q时,则将训练好的目标生成器和目标判别器作为训练好的条件生成对抗网络,若q<q时,则令q=q 1,执行步骤s6021、s6022。
[0114]
参照图7,本技术提供的步骤s6021一种实施例的方法流程图,该步骤s6021包括下述步骤:
[0115]
s701、将舌象训练集中的人体舌象图像和成对的真实标签进行拼接后,得到真实舌象图像并输入至当前生成器,计算当前生成器的生成损失值,基于生成损失值反向传播训练当前生成器,以输出第一预测标签;
[0116]
s702、将第一预测标签和对应的舌象训练集中的人体舌象图像进行拼接后,得到第一预测舌象图像;
[0117]
s703、将第一预测舌象图像和真实舌象图像输入至当前判别器进行预测,得到两个不同的第一预测结果,分别是与第一预测舌象图像对应的第一预测结果、与真实舌象图像对应的第一预测结果;
[0118]
s704、根据两个不同的第一预测结果,计算条件生成对抗网络的第一对抗损失值,采用梯度下降法并根据第一对抗损失值,对当前生成器的网络参数进行更新,以得到目标生成器。
[0119]
需要说明的是,第一预测标签是指当前生成器根据真实舌象图像输出的预测舌体图像;第一预测结果是指当前判别器根据第一预测舌象图像/真实舌象图像输出的真伪预测结果。
[0120]
参照图8,本技术提供的步骤s6022一种实施例的方法流程图,该步骤s6022包括下述步骤:
[0121]
s801、将真实舌象图像输入至目标生成器,得到第二预测标签;
[0122]
s802、将第二预测标签和对应的舌象训练集中的人体舌象图像进行拼接后,得到第二预测舌象图像;
[0123]
s803、将第二预测舌象图像和真实舌象图像输入当前判别器,计算当前判别器的判别损失值,基于判别损失值反向传播训练当前判别器,以输出两个不同的第二预测结果,分别是与第二预测舌象图像对应的第二预测结果、与真实舌象图像对应的第二预测结果;
[0124]
s804、根据两个不同的第二预测结果,计算条件生成对抗网络的第二对抗损失值,采用梯度下降法并根据第二对抗损失值,对当前判别器的网络参数进行更新后,得到目标判别器;
[0125]
需要说明的是,第二预测标签是指目标生成器根据真实舌象图像输出的预测舌体图像;第二预测结果是指当前判别器根据第二预测舌象图像/真实舌象图像输出的真伪判别结果。
[0126]
可选的,对训练好的条件生成对抗网络进行测试,具体地,准备测试数据集,将测试数据集中的舌象图像作为训练好的条件生成对抗网络1的输入进行分割,得到最后分割的舌体。
[0127]
如图9(a)为一张人体舌象图像,如图9(b)为u

net网络对图9(a)人体舌象图像进行分割后得到的舌体示意图,如图9(c)为基于u

net的条件生成对抗网络对图9(a)人体舌象图像进行分割后得到的舌体示意图,从中可以看出,通过两个方法的分割,可以大致分割出舌体图像,但使用u

net方法进行训练分割时,虽然也能大致分割出舌体的形状,但受到舌体旁边皮肤的影响,分割边界不够明确,会分割错误;而使用基于u

net的条件生成对抗网络时,可以较好分割出舌体的形状。
[0128]
本实施例通过的舌象分割方法通过vgg16网络提取舌象训练集的特征信息即舌体掩膜,作为网络训练的真实标签,并作为一种先验条件,将舌体掩膜集与舌象数据集一起输入条件生成对抗网络,便于对抗训练,引导条件生成对抗网络进行舌象分割,无需人工做标签,并且在舌象数据集小的情况下,也能够有效提供舌象分割精度。
[0129]
应理解,上述实施例中各步骤的序号的大小并不意味着执行顺序的先后,各过程
的执行顺序应以其功能和内在逻辑确定,而不应对本发明实施例的实施过程构成任何限定。
[0130]
本实施例还提供一种舌象分割系统,该舌象分割系统与上述实施例中舌象分割方法一一对应。如图10所示,该舌象分割系统包括数据获取模块1001、特征提取模块1002、训练模块1003和舌象分割模块1004。
[0131]
各功能模块详细说明如下:
[0132]
数据获取模块1001,用于获取舌象训练集,舌象训练集包括多张人体舌象图像;
[0133]
特征提取模块1002,用于对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;
[0134]
训练模块1003,用于将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;
[0135]
舌象分割模块1004,用于获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0136]
关于舌象分割系统的具体限定可以参见上文中对于舌象分割方法的限定,在此不再赘述。上述舌象分割系统中的各个模块可全部或部分通过软件、硬件及其组合来实现。上述各模块可以硬件形式内嵌于或独立于计算机设备中的处理器中,也可以以软件形式存储于计算机设备中的存储器中,以便于处理器调用执行以上各个模块对应的操作。
[0137]
本实施例还提供了一种计算机设备,该计算机设备可以是服务器,其内部结构图可以如图11所示。该计算机设备包括通过系统总线连接的处理器、存储器、网络接口和数据库。其中,该计算机设备的处理器用于提供计算和控制能力。该计算机设备的存储器包括非易失性存储介质、内存储器。该非易失性存储介质存储有操作系统、计算机程序和数据库。该内存储器为非易失性存储介质中的操作系统和计算机程序的运行提供环境。该计算机设备的数据库用于存储舌象数据集、真实标签集等数据。该计算机设备的网络接口用于与外部的终端通过网络连接通信。该计算机程序被处理器执行时以实现一种舌象分割方法,在一实施例中,处理器执行计算机程序时实现以下步骤:
[0138]
获取舌象训练集,舌象训练集包括多张人体舌象图像;
[0139]
对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;
[0140]
将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;
[0141]
获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0142]
本实施例还提供了一种计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时实现以下步骤:
[0143]
获取舌象训练集,舌象训练集包括多张人体舌象图像;
[0144]
对舌象训练集中的人体舌象图像进行特征提取和特征融合后,得到舌体掩膜集并作为真实标签集;
[0145]
将舌象训练集和成对的真实标签集输入至预设的条件生成对抗网络,通过迭代对抗训练,分别更新条件生成对抗网络的生成器和判别器,以得到训练完成的条件生成对抗网络;
[0146]
获取待识别舌象图像,并输入至训练完成的条件生成对抗网络进行舌象分割,生成舌体图像。
[0147]
本领域技术人员可以理解,实现上述实施例方法的全部或部分流程,可以通过计算机程序来指令相关的硬件来完成,所述的程序可存储于计算机可读存储介质中。其中,所述计算机可读存储介质为磁盘、光盘、只读存储记忆体或随机存储记忆体等。
[0148]
以上所述,仅为本发明较佳的具体实施方式,但本发明的保护范围并不局限于此,任何熟悉本技术领域的技术人员在本发明揭露的技术范围内,可轻易想到的变化或替换,都应涵盖在本发明的保护范围之内。
再多了解一些

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

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

相关文献

  • 日榜
  • 周榜
  • 月榜