发明公开
- 专利标题: 跨域场景下的卷烟缺陷检测模型训练方法及装置
-
申请号: CN202410609165.2申请日: 2024-05-16
-
公开(公告)号: CN118379581A公开(公告)日: 2024-07-23
- 发明人: 朱立明 , 章强 , 张博 , 陈明 , 方利梅 , 徐鑫 , 胡宏涛
- 申请人: 浙江中烟工业有限责任公司
- 申请人地址: 浙江省杭州市上城区中山南路77号
- 专利权人: 浙江中烟工业有限责任公司
- 当前专利权人: 浙江中烟工业有限责任公司
- 当前专利权人地址: 浙江省杭州市上城区中山南路77号
- 代理机构: 北京维澳知识产权代理有限公司
- 代理商 陈变花
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/80 ; G06V10/764 ; G06V10/766
摘要:
本申请公开了一种跨域场景下的卷烟缺陷检测模型训练方法及装置,方法包括:构建源域目标检测数据集和新增域目标检测数据集;构建教师模型,利用源域目标检测数据集对教师模型进行训练,训练完成后冻结教师模型的第一参数;构建网络结构与教师模型相同的学生模型,利用第一参数对学生模型进行初始化;利用新增域目标检测数据集对学生模型进行训练,训练过程中,基于类敏感度的模型参数更新方法对学生模型进行参数更新。本申请通过知识蒸馏的方式将原卷烟牌号和/或原缺陷类别的源域参数迁移到新模型中,并在对新模型的增量缺陷类别训练中对不同类敏感度的模型参数分别进行更新,实现了对新增牌号和/或新增缺陷类别的动态学习。