摘要
本公开提供了一种图像分类模型的训练方法,可以应用于计算机技术领域。该方法包括:确定教师模型和待训练的学生模型,所述教师模型为基于特征金字塔的分割模型,所述待训练的学生模型为分割模型;利用所述教师模型和图像样本训练集,对所述待训练的学生模型进行训练,得到训练完成的学生模型,所述图像样本训练集包括具有至少一种缺陷的图像训练样本,至少部分所述图像训练样本标记有真实标签;将所述训练完成的学生模型确定为图像分类模型,所述图像分类模型用于对具有缺陷的图像样本按照所述缺陷进行分类。本公开还提供了一种图像分类方法及相关设备。
技术关键词
图像分类模型
特征金字塔
图像训练样本
学生
教师
图像分类方法
训练集
变量
图像分类装置
模型训练模块
标签
层级
计算机程序产品
处理器
训练装置
指令
标记
可读存储介质