摘要
基于缺陷检测模型的缺陷检测方法和装置、计算机程序产品,其中缺陷检测模型包括经预训练的编码器、实例特征增强模块和解码器,编码器有一层或多层特征提取层,至少一层特征提取层之后嵌入有一个实例特征增强模块;该方法包括:将待检测图像输入编码器经各特征提取层及实例特征增强模块后得到缺陷特征图;将各层缺陷特征图输入解码器得到待检测图像的缺陷检测结果;其中每层特征提取层对其输入特征图进行特征提取以得到该层的缺陷特征图,若该层嵌入有实例特征增强模块,则还通过实例特征增强模块利用缺陷实例的语义信息对所提取的缺陷特征图进行特征增强,得到最终的缺陷特征图。从而能弥补预训练域与目标域之间的数据分布差异,提升检测性能。
技术关键词
缺陷检测方法
多层感知机
语义
注意力
编码器
缺陷特征提取
缺陷检测装置
矩阵
Softmax函数
计算机程序产品
输入解码器
图像获取模块
缺陷预测
通道
数据分布
参数
物体