公开(公告)号：CN118396947A

公开(公告)日：2024-07-26

申请号：CN202410484467.1

申请日：2024-04-22

Applicant: 河钢数字技术股份有限公司 ,  哈尔滨工业大学(深圳)(哈尔滨工业大学深圳科技创新研究院)

Inventor： 李毅仁 ,  明勇杰 ,  王凯军 ,  魏晓飞 ,  何勇军 ,  彭晶 ,  来博文 ,  苏敬勇

IPC: G06T7/00 ,  G06N3/042 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/096

Abstract: 本发明公开了一种适合边缘设备部署的钢材微小缺陷检测方法，涉及工业环境的钢材微小缺陷识别领域，通过将钢材缺陷图像压缩处理成规定尺寸的图像文件，该图像输入到基于U形结构的特征提取网络中，特征提取网络的可变形卷积与注意力模块负责对特征进行提取，注意力机制点积方式获取钢材缺陷中的上下文信息特征，同时为保留钢材缺陷某些初始的特征，将提取局部显著特征与全局上下文信息特征进行细粒度融合得到最终输出特征，将最终输出特征传递到解码器部分，由解码器进行特征映射并最终输出钢材微小缺陷检测的结果，为便于特征提取网络在边缘设备端部署，提出一种知识蒸馏学习框架，通过教师网络的学习，不断优化目标函数，达到对学生网络的指导。