【技术实现步骤摘要】
图像分类方法、装置、电子设备及存储介质
本专利技术涉及人工智能
,尤其涉及一种图像分类方法、装置、电子设备及计算机可读存储介质。
技术介绍
深度学习时代到来后,图像分类领域出现了一系列的突破,然而,这种突破是建立在有着大规模有标注数据集的基础上的。而大规模有标注数据集,则是一种奢侈。在类如医学图像领域,图像标注往往要花费领域专家的大量时间。专家间的认知差异也会导致标注出现噪声,而为了消除这种噪声则需要更多的专家进行盲标注。使用半监督学习方法可以减少标注负担。半监督方法可以运用大量无标注的数据和少量有标注数据来加强模型的性能。但是,半监督学习中一个重要的目标是要避免网络在小规模的有标注数据上出现过拟合。而实现这一目标的一个常用的必要假设则是网络拟合的函数的平滑性和一致性,即流形中相近的点应预测出相同的标注。举例来说,self-ensembling对在有局部扰动的无标注数据上出现不一致的预测进行惩罚,而对抗学习通过强制对不同的有着对抗性扰动的输入的预测相同来维持一致性。但是这些方法只考虑到了围绕单个数据点的扰动,即只对单数据点局部的平滑性做出了规范,并没有考虑到对不同数据点间的网络的全局平滑性和一致性进行约束,使得网络性能较低,导致图像分类准确性较低。
技术实现思路
本专利技术提供一种图像分类方法、装置、电子设备及计算机可读存储介质,其主要目的在于解决图像分类准确性较低的问题。为实现上述目的,本专利技术提供的一种图像分类方法,包括:获取原始图像集,其中,所述原始图像集中包 ...
【技术保护点】
1.一种图像分类方法,其特征在于,所述方法包括:/n获取原始图像集,其中,所述原始图像集中包括有标注图像集及无标注图像集;/n对所述无标注图像集中的图像进行标注猜测,得到标注猜测图像集,汇总所述标注猜测图像集及所述有标注图像集,得到图像训练集;/n在预设的图像分类网络中构建表征混合层,得到混合图像分类网络;/n利用所述图像训练集对所述混合图像分类网络进行训练,得到标准图像分类网络;/n利用所述标准图像分类网络对待分类图像进行分类,得到图像分类结果。/n
【技术特征摘要】
1.一种图像分类方法,其特征在于,所述方法包括:
获取原始图像集,其中,所述原始图像集中包括有标注图像集及无标注图像集;
对所述无标注图像集中的图像进行标注猜测,得到标注猜测图像集,汇总所述标注猜测图像集及所述有标注图像集,得到图像训练集;
在预设的图像分类网络中构建表征混合层,得到混合图像分类网络;
利用所述图像训练集对所述混合图像分类网络进行训练,得到标准图像分类网络;
利用所述标准图像分类网络对待分类图像进行分类,得到图像分类结果。
2.如权利要求1所述的图像分类方法,其特征在于,所述对所述无标注图像集中的图像进行标注猜测,得到标注猜测图像集,包括:
利用预构建的生成对抗网络,对所述无标注图像集中的图像进行预设次数的数据增强处理,得到增强图像;
利用预设的标注猜测公式对所述增强图像进行标注猜测,得到标注猜测图像;
汇总所述标注猜测图像得到所述标注猜测图像集。
3.如权利要求2所述的图像分类方法,其特征在于,所述利用预设的标注猜测公式对所述增强图像进行标注猜测,得到标注猜测图像,包括:
利用下述猜测公式对所述增强图像进行标注猜测,得到标注猜测图像:
其中,qb表示无标注图像的猜测标注,ub表示无标注图像集,ub,m表示无标注图像集中第m个图像,f(ub,m;θ)表示参数θ的网络拟合函数对ub的第m个图像为输入的输出,M表示数据增强的次数。
4.如权利要求1所述的图像分类方法,其特征在于,所述在预设的图像分类网络中构建表征混合层,得到混合图像分类网络,包括:
在所述图像分类网络中随机选取一层作为表征混合层;
将所述表征混合层之前的网络设置为编码层,及将所述表征混合层之后的网络设置为解码层;
汇总所述编码层、所述表征混合层及所述解码层,得到所述混合图像分类网络。
5.如权利要求4所述的图像分类方法,其特征在于,所述利用所述图像训练集对所述混合图像分类网络进行训练,得到标准图像分类网络,包括:
利用所述混合图像分类网络中的编码层对所述图像训练集中图像对应的标注进行表征编码,得到隐表征对;
利用所述混合图像分类网络中的表征混合层对所述隐表征对进行线性混合,得到线性混合隐表征对;
根据所述线性混合隐表征对计算损失值,当所述损失值小于预设的损失阈值时,得到所述标准图像分类网络。
6.如权利要求5所述的图像分类方法,其特征在于,所述利用所述混合图像分类网络中的表征混合层对所述隐表征对进...
【专利技术属性】
技术研发人员:刘杰,王健宗,瞿晓阳,
申请(专利权)人:平安科技深圳有限公司,
类型:发明
国别省市:广东;44
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。