首页/ 征图新视（江苏）科技股份有限公司/ 专利详情

基于知识蒸馏的正样本工业缺陷检测方法

基本信息

申请号 CN202110415805.2 申请日 -
公开（公告）号 CN112991330B 公开（公告）日 2021-08-13
申请公布号 CN112991330B 申请公布日 2021-08-13
分类号 G06T7/00（2017.01）I;G06N3/08（2006.01）I;G01N21/88（2006.01）I 分类 计算；推算；计数;
发明人 和江镇;王岩松;都卫东;方志斌;夏海峰 申请（专利权）人 征图新视（江苏）科技股份有限公司
代理机构 常州品益专利代理事务所（普通合伙） 代理人 王涵江
地址 213161江苏省常州市武进经济开发区锦华路258-6号
法律状态 -

摘要

摘要 本发明涉及一种基于知识蒸馏的正样本工业缺陷检测方法，包括步骤（S1）：构建工业数据集。步骤（S2）：预处理操作，预处理后的工业数据集包括正样本集和无标注的缺陷样本集。步骤（S3）：使用自监督对比学习在步骤（S2）形成的工业数据集上预训练教师网络模型；步骤（S4）：在步骤（S2）形成的正样本集的基础上，利用步骤（S3）训练得到的教师网络模型指导学生网络模型训练。步骤（S5）：使用步骤（S3）训练得到的教师网络模型以及步骤（S4）训练得到的学生网络模型对待测图片进行缺陷检测。由于学生网络模型只学习到提取正样本特征的能力，对缺陷区域提取的特征与教师网络模型存在较大差异，从而可以作为缺陷判断的依据。