-
公开(公告)号:CN116863207A
公开(公告)日:2023-10-10
申请号:CN202310750086.9
申请日:2023-06-21
Applicant: 浙江大学
IPC: G06V10/764 , G06V10/774 , G06V10/82
Abstract: 本发明公开了一种基于知识蒸馏的图像分类方法,包括:(1)构建一个深度神经网络作为第一深度模型;(2)将构建的第一深度模型在图像训练集上进行训练;(3)提取训练好的第一深度模型中的网络骨干部分,在每层的输出后加入辅助分类头,随机初始化辅助分类头,构成第二深度模型;(4)冻结网络骨干部分,训练辅助分类头;(5)构建一个包含每层辅助分类头的深度神经网络,进行随机初始化,构成第三深度模型;(6)逐层训练第三深度模型,利用第二深度模型每层辅助分类头的输出作为监督;(7)提取第三深度模型的骨干部分以及最后一层辅助分类头,进行图像分类。利用本发明,可以提升训练模型的优化能力,提高图像分类的效果。