摘要
本发明公开了一种基于生成对抗网络的工业成品微缺陷检测方法,包括如下步骤:S1、采集工业成品表面的高分辨率图像,构建正常图像与缺陷图像的数据集;S2、提取图像特征向量并与缺陷类型向量拼接,输入至含几何结构变换模块的生成器网络,生成缺陷候选图像;S3、将缺陷候选图像输入含拓扑结构分析模块的判别器网络,输出真实性判别结果、结构连通性特征指标和缺陷类型预测标签;S4、联合训练生成器和判别器网络;S5、输入待检测图像,获取缺陷候选图像和预测结果;S6、生成缺陷概率图、显著性图和二值掩膜,定位缺陷区域;S7、结合预测标签、显著性图和掩膜,输出微缺陷检测和分类结果。本发明提升了复杂背景下微缺陷的检测精度与分类能力。
技术关键词
生成器网络
缺陷检测方法
生成对抗网络
图像特征向量
图像解码单元
特征提取单元
图像编码
指标
掩膜
标签
成品
分析模块
工业
定位缺陷
多层卷积神经网络
像素
图像增强模块
双曲正切函数