发明公开
- 专利标题: 基于重参化和注意力机制的贴片保险丝表面缺陷检测方法及装置
-
申请号: CN202311185801.5申请日: 2023-09-14
-
公开(公告)号: CN117455832A公开(公告)日: 2024-01-26
- 发明人: 周琪 , 王鸿鹏 , 程正涛
- 申请人: 哈尔滨工业大学(深圳)(哈尔滨工业大学深圳科技创新研究院)
- 申请人地址: 广东省深圳市南山区桃源街道深圳大学城哈尔滨工业大学校区
- 专利权人: 哈尔滨工业大学(深圳)(哈尔滨工业大学深圳科技创新研究院)
- 当前专利权人: 哈尔滨工业大学(深圳)(哈尔滨工业大学深圳科技创新研究院)
- 当前专利权人地址: 广东省深圳市南山区桃源街道深圳大学城哈尔滨工业大学校区
- 代理机构: 广州市华学知识产权代理有限公司
- 代理商 李斌
- 主分类号: G06T7/00
- IPC分类号: G06T7/00 ; G06V10/764 ; G06V10/774 ; G06V10/82 ; G06V10/80 ; G06N3/0464 ; G06N3/084 ; G06N5/04
摘要:
本发明公开一种基于重参化和注意力机制的贴片保险丝表面缺陷检测方法及装置,方法包括获取已标注贴片保险丝的表面图像并形成总训练集;基于总训练集构建并训练缺陷检测模型;将训练好的缺陷检测模型重参化以转化为推理模型;利用NVIDIA TensorRT库优化训练好的缺陷检测模型,进行格式转化并保存;检测并分类待测贴片保险丝的表面图像。本发明提出了一种全新的参数量较小且训练速度、推理速度较快与准确率高的基于重参化与注意力机制的深度学习模型,并将其应用于贴片保险丝缺陷检测任务上,提出一套完整的保险丝缺陷检测流程。