(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210489486.4 (22)申请日 2022.05.06 (71)申请人 讯飞智元信息科技有限公司 地址 230088 安徽省合肥市高新区望江西 路666号讯飞大厦8层-10层 申请人 科大讯飞股份有限公司 (72)发明人 周晨霞　谭昶　张友国　吕军　 贾若然　姜殿洪　刘江　冯祥　 韩辉　李亚玲　 (74)专利代理 机构 北京路浩知识产权代理有限 公司 11002 专利代理师 程琛 (51)Int.Cl. G06V 10/764(2022.01) G06V 10/80(2022.01)G06V 10/82(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 图像分类方法、 装置、 电子设备与存 储介质 (57)摘要 本发明提供一种图像分类方法、 装置、 电子 设备与存储介质， 其中方法包括： 确定待分类的 图像； 基于分类模型， 对所述图像进行多属性分 类， 得到所述图像的多属性分类结果； 所述分类 模型是在预训练编码器的基础上， 基于第一样本 图像及其对应的样本多属性分类结果训练得到 的； 所述预训练编码器是基于多个图像处理任务 进行自监督学习得到的模型中的编码器。 本发明 提供的方法、 装置、 电子设备与存储介质， 训练流 程简单， 方法复用性强， 节省了数据的标注成本， 同时极大提升了模型在多属性分类任务上的分 类精度， 在此基础上， 应用分类模型对输入的图 像进行多属性分类， 实现了通过同一网络模型识 别出图像的不同属性。 权利要求书2页 说明书12页 附图5页 CN 115063625 A 2022.09.16 CN 115063625 A 1.一种图像分类方法， 其特 征在于， 包括： 确定待分类的图像； 基于分类模型， 对所述图像进行多属性分类， 得到所述图像的多属性分类结果； 所述分类模型是在预训练编码器的基础上， 基于第 一样本图像及其对应的样本多属性 分类结果训练得到的； 所述预训练编码器是基于多个图像处 理任务进行自监 督学习得到的模型中的编码器。 2.根据权利要求1所述的图像分类方法， 其特征在于， 所述预训练编码器基于如下步骤 确定： 基于第二样本 图像， 确定所述多个图像处理任务分别对应的输入样本 图像和伪标签， 所述多个图像处 理任务包括图像生成任务和/或图像判别任务； 基于所述多个图像处理任务分别对应的输入样本 图像和伪标签， 进行模型训练， 得到 所述预训练编码器。 3.根据权利要求2所述的图像分类方法， 其特征在于， 所述基于所述多个图像处理任务 分别对应的输入样本图像和伪标签， 进行模型训练， 得到所述预训练编码器， 包括： 基于所述多个图像处理任务中的上一图像处理任务对应的预训练模型中的编码器， 确 定当前图像处 理任务对应的初始模型； 基于所述当前图像处理任务对应的输入样本图像和伪标签， 对当前图像处理任务对应 的初始模型进行训练， 得到所述当前图像处理任务对应的预训练模型， 并将所述当前图像 处理任务更新 为下一图像处 理任务； 基于所述多个图像处理任务中的最后一个图像处理任务对应的预训练模型中的编码 器， 确定所述预训练编码器。 4.根据权利要求2所述的图像分类方法， 其特征在于， 所述图像生成任务包括图像着 色、 图像修复和图像超分辨率中的至少一种， 所述图像判别任务包括图像拼图和/或图像旋 转。 5.根据权利要求1至4中任一项所述的图像分类方法， 其特征在于， 所述分类模型的损 失函数是基于所述多属 性分类的各属 性分类任务在训练过程中的各步对应的损失值和各 步对应的权 重确定。 6.根据权利要求5所述的图像分类方法， 其特征在于， 任一属性分类任务在任一步对应 的权重基于所述各属性分类任务在所述任一 步的前一 步对应的训练速度确定 。 7.根据权利要求1至4中任一项所述的图像分类方法， 其特征在于， 所述分类模型基于 如下步骤确定： 基于所述预训练编码器， 以及所述多属性分类的各属性分类任务对应的分类器， 构建 初始分类模型； 基于所述第 一样本图像及其对应的样本多属性分类结果， 对所述初始分类模型进行训 练， 得到中间分类模型； 基于第三样本图像及其对应的样本多属性分类结果， 对所述中间分类模型中的各分类 器进行训练， 得到所述分类模型。 8.一种图像分类装置， 其特 征在于， 包括： 图像确定单 元， 用于确定待分类的图像；权　利　要　求　书 1/2 页 2 CN 115063625 A 2多属性分类单元， 用于基于分类模型， 对所述图像进行多属性分类， 得到所述图像的多 属性分类结果； 所述分类模型是在预训练编码器的基础上， 基于第 一样本图像及其对应的样本多属性 分类结果训练得到的； 所述预训练编码器是基于多个图像处 理任务进行自监 督学习得到的模型中的编码器。 9.一种电子设备， 包括存储器、 处理器及存储在所述存储器上并可在所述处理器上运 行的计算机程序， 其特征在于， 所述处理器执行所述程序时实现如权利要求1至7任一项所 述图像分类方法。 10.一种非暂态计算机可读存储介质， 其上存储有计算机程序， 其特征在于， 所述计算 机程序被处 理器执行时实现如权利要求1至7任一项所述图像分类方法。权　利　要　求　书 2/2 页 3 CN 115063625 A 3