-
公开(公告)号:CN119516249A
公开(公告)日:2025-02-25
申请号:CN202411464089.7
申请日:2024-10-21
Applicant: 华南农业大学
IPC: G06V10/764 , G06V10/82 , G06N3/0455 , G06V10/774 , G06V10/776 , G06V10/40 , G06V10/30 , G06V10/75 , G06T7/00
Abstract: 本发明公开了一种基于变分自编码器与去噪网络的多类别缺陷检测方法及装置,方法包括:获取工业产品表面缺陷图像并进行预处理,构建训练集和验证集;方法的训练过程分为两个阶段:第一阶段,将预处理后的训练图像输入变分自编码器进行训练,学习图像的特征分布,并通过重建图像来捕捉正常图像的特征。第二阶段,利用去噪网络与特征引导模块在变分自编码器的特征空间中对图像特征进行二次重建,从而增强模型的重建能力和多类别重建能力;在推理阶段,通过特征提取器提取原始图像与重建图像的多尺度特征,进行特征比对,生成异常评分图。本发明在增强变分自编码器重建能力的同时保留正常信息,有效减少重建失败而导致的缺陷漏检。