图像识别方法、图像识别装置、电子设备、存储介质制造方法及图纸

技术编号:38368753 阅读:18 留言:0更新日期:2023-08-05 17:33
本申请公开了一种图像识别方法、图像识别装置、电子设备、存储介质,本申请涉及人工智能的图像识别技术领域。图像识别方法包括:根据预设的原始图像识别模型对第一样本图像进行图像识别,得到第一图像标签;根据第一样本图像得到第一差异图像;根据原始图像识别模型对第一差异图像进行图像识别,得到第二图像标签,以此对原始图像识别模型进行参数调整,得到初步图像识别模型;构建图像数据集;根据初步图像识别模型对第二样本图像进行图像识别,得到第三图像标签;对初步图像识别模型进行参数调整,得到目标图像识别模型;根据目标图像识别模型对待识别图像进行图像识别。本实施例的图像识别方法能够提高对多种对象类型图像的识别准确性。的识别准确性。的识别准确性。

【技术实现步骤摘要】
图像识别方法、图像识别装置、电子设备、存储介质


[0001]本申请涉及人工智能的图像识别
,尤其是涉及一种图像识别方法、图像识别装置、电子设备、存储介质。

技术介绍

[0002]相关技术中,通过图像识别模型对图像进行识别,以判断待识别图像是否属于正常类别。但由于相关技术中图像识别模型在训练时仅采用单一对象类型的样本图像进行训练,因此该模型仅能够保证对相同对象类型的图像的识别准确性。也就是说,在获取到其他对象类型的图像时,相关技术中的图像识别模型容易出现识别准确性低的现象。因此,如何提高对多种对象类型图像的识别准确性成了亟待解决的技术问题。

技术实现思路

[0003]本申请旨在至少解决现有技术中存在的技术问题之一。为此,本申请提出一种图像识别方法,能够提高对多种类别图像的识别准确性。
[0004]本申请还提出一种图像识别装置,和一种应用上述图像识别方法的电子设备以及一种应用上述图像识别方法的计算机可读存储介质。
[0005]根据本申请的第一方面实施例的图像识别方法,包括:
[0006]根据预设的原始图像识别模型对获取的第一样本图像进行图像识别,得到第一图像标签;其中,所述第一图像标签用于表征所述第一样本图像的图像类别,所述图像类别包括正常类别和异常类别;
[0007]对所述第一样本图像进行数据增强处理,得到第一差异图像;其中,所述第一差异图像的图像类别与所述第一样本图像的图像类别不相同;
[0008]将所述第一差异图像输入至所述原始图像识别模型进行图像识别,得到第二图像标签;其中,所述第二图像标签用于表征所述第一差异图像的图像类别;
[0009]根据所述第一样本图像、所述第一图像标签、所述第一差异图像和所述第二图像标签对所述原始图像识别模型进行参数调整,得到初步图像识别模型;
[0010]根据所述第一样本图像、所述第一差异图像构建图像数据集;
[0011]根据所述图像数据集、所述初步图像识别模型对获取到的第二样本图像进行图像识别,得到第三图像标签;其中,所述第三图像标签用于表征所述第二样本图像的图像类别;
[0012]根据所述第二样本图像、所述图像数据集、所述第三图像标签对所述初步图像识别模型进行参数调整,得到目标图像识别模型;
[0013]根据所述目标图像识别模型对获取到的待识别图像进行图像识别。
[0014]根据本申请实施例的图像识别方法,至少具有如下有益效果:根据预设的原始图像识别模型对获取的第一样本图像进行图像识别,得到第一图像标签。对第一样本图像进行数据增强处理,得到第一差异图像。将第一差异图像输入至所述原始图像识别模型进行
图像识别,得到第二图像标签。根据第一样本图像、第一图像标签、第一差异图像和第二图像标签对原始图像识别模型进行参数调整,得到初步图像识别模型。根据第一样本图像、第一差异图像构建图像数据集。根据图像数据集、初步图像识别模型对获取到的第二样本图像进行图像识别,得到第三图像标签。根据第二样本图像、图像数据集、第三图像标签对初步图像识别模型进行参数调整,得到目标图像识别模型。最后,根据得到的目标图像识别模型对获取到的待识别图像进行图像识别。本实施例的图像识别方法能够得到与样本图像的类别不相同的差异图像,将差异图像和样本图像一起对图像识别模型进行训练,并将差异图像和样本图像存储为图像数据集,以用于后续对图像识别模型的训练,避免了图像识别模型对历史样本图像的遗忘,以此提高了对多种对象类型图像的识别准确性。
[0015]根据本申请的一些实施例,所述对所述第一样本图像进行数据增强处理,得到第一差异图像,包括:
[0016]对所述第一样本图像进行像素截取操作,得到图像像素数据;
[0017]获取所述第一样本图像的第一随机位置数据;
[0018]根据所述第一随机位置数据、所述图像像素数据更新所述第一样本图像,得到所述第一差异图像。
[0019]根据本申请的一些实施例,所述对所述第一样本图像进行数据增强处理,得到第一差异图像,包括:
[0020]获取所述第一样本图像的第二随机位置数据;
[0021]根据所述第二随机位置数据从所述第一样本图像中筛选得到样本像素数据;
[0022]对所述样本像素数据进行像素值调整操作,得到所述第一差异图像。
[0023]根据本申请的一些实施例,所述原始图像识别模型包括特征提取子模型和分类子模型;
[0024]所述根据预设的原始图像识别模型对获取的第一样本图像进行图像识别,得到第一图像标签,包括:
[0025]将所述第一样本图像输入至所述特征提取子模型进行特征提取操作,得到样本特征向量;
[0026]将所述样本特征向量输入至所述分类子模型进行图像类别识别,得到所述第一图像标签。
[0027]根据本申请的一些实施例,所述特征提取子模型包括图像分割层和编码层;
[0028]所述将所述第一样本图像输入至所述特征提取子模型进行特征提取操作,得到样本特征向量,包括:
[0029]根据所述图像分割层对所述第一样本图像进行图像分割操作,得到样本切割图像;
[0030]根据所述编码层对所述样本切割图像进行归一化处理,得到所述样本特征向量。
[0031]根据本申请的一些实施例,所述根据所述第二样本图像、所述图像数据集、所述第三图像标签对所述初步图像识别模型进行参数调整,得到目标图像识别模型,包括:
[0032]对所述第二样本图像进行数据增强处理,得到第二差异图像;其中,所述第二差异图像的图像类别与所述第二样本图像的图像类别不相同;
[0033]将所述第二差异图像输入至所述初步图像识别模型进行图像识别,得到第四图像
标签;其中,所述第四图像标签用于表征所述第二差异图像的图像类别;
[0034]根据所述第二样本图像、所述第二差异图像、所述图像数据集、所述第三图像标签、所述第四图像标签对所述初步图像识别模型进行参数调整,得到所述目标图像识别模型。
[0035]根据本申请的一些实施例,所述根据所述第二样本图像、所述第二差异图像、所述图像数据集、所述第三图像标签、所述第四图像标签对所述初步图像识别模型进行参数调整,得到所述目标图像识别模型,包括:
[0036]根据所述第二样本图像、所述第三图像标签得到正常损失数据;
[0037]根据所述第二差异图像、所述第四图像标签得到异常损失数据;
[0038]根据所述图像数据集得到历史损失数据;
[0039]根据所述正常损失数据、异常损失数据、历史损失数据对所述初步图像识别模型进行参数调整,得到所述目标图像识别模型。
[0040]根据本申请的第二方面实施例的图像识别装置,包括:
[0041]第一图像识别模块,所述第一图像识别用于根据预设的原始图像识别模型对获取的第一样本图像进行图像识别,得到第一图像标签;其中,所述第一图像标签用于表征所述第一样本图像的图像类别,所述图像本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.图像识别方法,其特征在于,所述方法包括:根据预设的原始图像识别模型对获取的第一样本图像进行图像识别,得到第一图像标签;其中,所述第一图像标签用于表征所述第一样本图像的图像类别,所述图像类别包括正常类别和异常类别;对所述第一样本图像进行数据增强处理,得到第一差异图像;其中,所述第一差异图像的图像类别与所述第一样本图像的图像类别不相同;将所述第一差异图像输入至所述原始图像识别模型进行图像识别,得到第二图像标签;其中,所述第二图像标签用于表征所述第一差异图像的图像类别;根据所述第一样本图像、所述第一图像标签、所述第一差异图像和所述第二图像标签对所述原始图像识别模型进行参数调整,得到初步图像识别模型;根据所述第一样本图像、所述第一差异图像构建图像数据集;根据所述图像数据集、所述初步图像识别模型对获取到的第二样本图像进行图像识别,得到第三图像标签;其中,所述第三图像标签用于表征所述第二样本图像的图像类别;根据所述第二样本图像、所述图像数据集、所述第三图像标签对所述初步图像识别模型进行参数调整,得到目标图像识别模型;根据所述目标图像识别模型对获取到的待识别图像进行图像识别。2.根据权利要求1所述的图像识别方法,其特征在于,所述对所述第一样本图像进行数据增强处理,得到第一差异图像,包括:对所述第一样本图像进行像素截取操作,得到图像像素数据;获取所述第一样本图像的第一随机位置数据;根据所述第一随机位置数据、所述图像像素数据更新所述第一样本图像,得到所述第一差异图像。3.根据权利要求1所述的图像识别方法,其特征在于,所述对所述第一样本图像进行数据增强处理,得到第一差异图像,包括:获取所述第一样本图像的第二随机位置数据;根据所述第二随机位置数据从所述第一样本图像中筛选得到样本像素数据;对所述样本像素数据进行像素值调整操作,得到所述第一差异图像。4.根据权利要求1所述的图像识别方法,其特征在于,所述原始图像识别模型包括特征提取子模型和分类子模型;所述根据预设的原始图像识别模型对获取的第一样本图像进行图像识别,得到第一图像标签,包括:将所述第一样本图像输入至所述特征提取子模型进行特征提取操作,得到样本特征向量;将所述样本特征向量输入至所述分类子模型进行图像类别识别,得到所述第一图像标签。5.根据权利要求4所述的图像识别方法,其特征在于,所述特征提取子模型包括图像分割层和编码层;所述将所述第一样本图像输入至所述特征提取子模型进行特征提取操作,得到样本特征向量,包括:
根据所述图像分割层对所述第一样本图像进行图像分割操作,得到样本切割图像;根据所述编码层对所述样本切割图像进行归一化处理,得到所述样本特征向量。6.根据权利要求1至5任一项所述的图像识别方法,其特征在于,所述根据所述第二样本图像、所述图像数据集、所述第三图像标签对所述初步图像识别模型进行参数调整,得到目标图像识别模型,包括:对所述第二样本图像进行数据增强处理,得到第二差异图像;...

【专利技术属性】
技术研发人员:李武津詹佳伟高斌斌刘俊刘永郑锋
申请(专利权)人:南方科技大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1