图像分类模型训练的方法及装置、电子设备和存储介质

AITNT
正文
推荐专利
图像分类模型训练的方法及装置、电子设备和存储介质
申请号:CN202411043640
申请日期:2024-07-31
公开号:CN118887473A
公开日期:2024-11-01
类型:发明专利
摘要
本公开提供了一种图像分类模型训练的方法及装置、电子设备和存储介质,将训练图像集输入第一分类模型进行图像预测训练,并基于训练图像集中的训练标签图像进行有监督训练;将训练图像集输入第二分类模型进行图像预测训练;利用第一训练损失、有监督训练损失、第二训练损失,计算图像分类模型的模型损失函数,并基于模型损失函数的计算结果,对图像分类模型的模型参数进行优化,以获得训练完成的图像分类模型。与相关技术相比,本公开实施例通过将训练图像集分别输入第一分类模型、第二分类模型进行训练,实现半监督训练。这使得模型能够利用无标签数据进行学习,从而扩大训练数据的规模,挖掘更多潜在的信息,有助于提高模型的泛化能力和鲁棒性。
技术关键词
图像分类模型训练 投影变换矩阵 教师 学生 蒸馏 图像增强 半监督训练 无标签数据 电子设备 图像结构 解码器 计算机程序产品 处理器通信
系统为您推荐了相关专利信息
1
图像分割模型训练方法、装置、设备及存储介质
教师 置信度阈值 图像分割模型 实例分割 数据
2
一种基于决策树的不确定性再制造调度方法
序列 代表 教师 编码 利润
3
一种变压器状态检测方法及装置
绝缘油溶解气体 深度卷积神经网络 数据 变压器 状态检测方法
4
一种基于自蒸馏的低码率半参考图像质检方法及系统
语义向量 质检方法 样本 图像编码器 教师
5
跨届成绩的预测方法、装置、电子设备及存储介质
二进制特征 XGBoost模型 样本 学生 成绩
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号