3 个月前

FAIR:面向工业视觉异常检测的频域感知图像修复

FAIR:面向工业视觉异常检测的频域感知图像修复

摘要

基于图像重建的异常检测模型在工业视觉检测中得到了广泛研究。然而,现有模型通常面临正常样本重建保真度与异常样本重建可区分性之间的权衡问题,这一矛盾会损害整体检测性能。本文发现,通过利用正常与异常重建误差在频域上的显著差异,可以更有效地缓解上述权衡。为此,我们提出了一种新颖的自监督图像修复任务——频率感知图像修复(Frequency-aware Image Restoration, FAIR),该方法仅从图像的高频分量中恢复原始图像。FAIR能够在精确重建正常模式的同时,有效抑制对异常区域的不良泛化。仅使用一个简单的基础UNet结构,FAIR在多个缺陷检测数据集上均实现了当前最优的检测性能,且具有更高的计算效率。代码已开源:https://github.com/liutongkun/FAIR。

代码仓库

liutongkun/fair
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-mvtec-adFAIR
Detection AUROC: 98.6
Segmentation AUPRO: 94.0
Segmentation AUROC: 98.2
anomaly-detection-on-visaFAIRnoDTD
Detection AUROC: 97.1
Segmentation AUPRO (until 30% FPR): 91.2
Segmentation AUROC: 98.7

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
FAIR:面向工业视觉异常检测的频域感知图像修复 | 论文 | HyperAI超神经