发明公开
- 专利标题: 一种基于动量对比学习的工业图像缺陷检测方法
-
申请号: CN202410317921.4申请日: 2024-03-20
-
公开(公告)号: CN118037708A公开(公告)日: 2024-05-14
- 发明人: 郑瑞娟 , 王雪琦 , 张明川 , 刘牧华 , 刘铭 , 冀治航 , 吴庆涛 , 朱军龙
- 申请人: 河南科技大学
- 申请人地址: 河南省洛阳市洛龙区开元大道263号
- 专利权人: 河南科技大学
- 当前专利权人: 河南科技大学
- 当前专利权人地址: 河南省洛阳市洛龙区开元大道263号
- 代理机构: 郑州中科鼎佳专利代理事务所
- 代理商 田惠玲
- 主分类号: G06T7/00
- IPC分类号: G06T7/00 ; G06T7/11 ; G06V10/74 ; G06N3/09
摘要:
一种基于动量对比学习的工业图像缺陷检测方法,涉及视觉检测技术领域,包含以下步骤:S1、初始化编码器#imgabs0#和#imgabs1#,初始化队列#imgabs2#;S2、对初始图像进行数据增广,获得查询样本#imgabs3#、正样本#imgabs4#和负样本#imgabs5#,#imgabs6#,#imgabs7#;S3、通过编码器#imgabs8#得到特征#imgabs9#;通过编码器#imgabs10#得到特征#imgabs11#,#imgabs12#,#imgabs13#和#imgabs14#;S4、将#imgabs15#,#imgabs16#,#imgabs17#加入队列#imgabs18#;S5、分别对#imgabs19#与#imgabs20#和#imgabs21#进行计算,得到logits和labels;S6、对logits和labels计算InfoNCE得到损失;S7、由对比损失更新编码器#imgabs22#参数;S8、动量更新编码器#imgabs23#的参数;S9、对S2~S8进行循环,得到最佳模型;S10、选择最佳模型进行图像检测;本发明能够用于工业生产制造场景中对产品的质量检测,结合自监督学习建立了基于动量对比学习的工业图像缺陷检测模型。