Invention Publication
- Patent Title: 图像分类模型训练方法、图像分类方法及相关装置
-
Application No.: CN202210626206.XApplication Date: 2022-06-02
-
Publication No.: CN114972791APublication Date: 2022-08-30
- Inventor: 邓佳丽 , 丁子霖 , 刘明 , 龚海刚 , 王晓敏 , 刘明辉 , 程旋 , 解天舒
- Applicant: 电子科技大学长三角研究院(衢州)
- Applicant Address: 浙江省衢州市柯城区芹江东路288号1幢18楼
- Assignee: 电子科技大学长三角研究院(衢州)
- Current Assignee: 电子科技大学长三角研究院(衢州)
- Current Assignee Address: 浙江省衢州市柯城区芹江东路288号1幢18楼
- Agency: 北京超凡宏宇专利代理事务所
- Agent 张欣欣
- Main IPC: G06V10/44
- IPC: G06V10/44 ; G06V10/764 ; G06V10/774 ; G06V10/82 ; G06N3/04

Abstract:
本发明的实施例提供了一种图像分类模型训练方法、图像分类方法及相关装置,涉及图像处理领域。首先,获取原始图像及原始图像的类型标签;将原始图像输入预先构建的图像分类模型,图像分类模型包括特征提取网络和类型预测网络,特征提取网络包括N个依次串联的卷积层;再利用特征提取网络对原始图像进行特征提取,得到第N个所述卷积层输出的第一特征图和第N‑1个卷积层输出的第二特征图;基于第一特征图、类型标签和预设的多个二值掩码,生成第一训练集,基于第二特征图、类型标签和多个二值掩码,生成第二训练集;最后,利用第一训练集和第二训练集对类型预测网络进行训练,得到训练后的图像分类模型,从而减少模型训练过程产生的额外开销。
Public/Granted literature
- CN114972791B 图像分类模型训练方法、图像分类方法及相关装置 Public/Granted day:2025-04-25
Information query