3 个月前

基于全卷积跨尺度流的基于图像的缺陷检测

基于全卷积跨尺度流的基于图像的缺陷检测

摘要

在工业制造过程中,缺陷往往在不可预测的时间以未知的形式出现。针对这一问题,我们提出了一种无需任何缺陷样本图像即可实现自动缺陷检测的新方法。现有研究通常通过强统计先验或过于简化的数据表示来建模无缺陷图像数据的分布,而我们的方法则能够处理细粒度的特征表示,同时融合图像的全局与局部上下文信息,并灵活地估计数据密度。为此,我们提出了一种新型的全卷积跨尺度归一化流(Cross-scale Normalizing Flow, CS-Flow),该模型可联合处理多尺度的特征图。利用归一化流为输入样本分配有意义的概率似然值,实现了高效的图像级缺陷检测。此外,由于归一化流保留了空间结构,其隐空间具有可解释性,从而能够精确定位图像中的缺陷区域。在基准数据集Magnetic Tile Defects和MVTec AD上的实验表明,我们的方法在图像级缺陷检测任务中达到了新的最先进水平,在15个类别中的4个类别上实现了100%的AUROC。

代码仓库

marco-rudolph/cs-flow
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-mvtec-adCS-Flow
Detection AUROC: 98.7
anomaly-detection-on-surface-defect-saliencyCS-Flow (unsupervised)
Detection AUROC: 99.3

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于全卷积跨尺度流的基于图像的缺陷检测 | 论文 | HyperAI超神经