一种基于多尺度特征蒸馏的缺陷检测方法、装置及介质

基本信息

申请号 CN202111031303.6 申请日 -
公开(公告)号 CN113962928A 公开(公告)日 2022-01-21
申请公布号 CN113962928A 申请公布日 2022-01-21
分类号 G06T7/00(2017.01)I;G06V10/74(2022.01)I;G06K9/62(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 张重阳;姚欣成;张保柱;刘振宇 申请(专利权)人 宁波海棠信息技术有限公司
代理机构 宁波市鄞州盛飞专利代理事务所(特殊普通合伙) 代理人 龙洋
地址 315040浙江省宁波市高新区甬江大道2660号新材料国际创新中心东区A2-1102
法律状态 -

摘要

摘要 本发明属于图像缺陷检测技术领域,提供了一种基于多尺度特征蒸馏的缺陷检测方法、装置及介质,包括步骤:构建基于多尺度特征蒸馏的缺陷检测模型,所述检测模型包括老师网络和学生网络;对所述缺陷检测模型进行训练,利用所述学生网络通过特征蒸馏机制学习老师网络的知识,得到训练后的缺陷检测模型;通过训练后的缺陷检测模型对检测样本进行缺陷检测,判定检测样本中的缺陷品以及缺陷区域。本发明的优点在于通过输出多个尺度下的特征图,算法可以检测不同尺度的缺陷,并且由于本方法主要在于运行两次前向传播,该方法可以达到实时检测的要求,更加适合于工业缺陷检测等场景中的应用。