3 个月前

SuperSimpleNet:面向快速可靠表面缺陷检测的无监督与有监督学习统一方法

SuperSimpleNet:面向快速可靠表面缺陷检测的无监督与有监督学习统一方法

摘要

表面缺陷检测的目标是识别并定位所捕获物体表面的异常区域,这一任务在多个行业中日益受到重视。当前的方法往往难以满足工业领域对高性能、高一致性、快速运行以及充分利用全部可用训练数据的广泛需求。为弥补这些不足,我们提出了SuperSimpleNet,一种基于SimpleNet演进而来的创新判别模型。该模型在训练一致性、推理速度和检测性能方面均显著优于其前身。SuperSimpleNet可在仅使用正常样本进行无监督训练的情况下运行,同时在有标注异常样本可用时,也能有效利用这些信息提升性能。在四个具有挑战性的基准数据集上的实验结果表明,SuperSimpleNet在有监督和无监督两种设置下均达到了当前最优水平。代码已开源:https://github.com/blaz-r/SuperSimpleNet。

代码仓库

blaz-r/supersimplenet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-mvtec-adSuperSimpleNet
Detection AUROC: 98.4
FPS: 107 (Tesla V100S)
Segmentation AUPRO: 91.1
anomaly-detection-on-visaSuperSimpleNet
Detection AUROC: 93.4
Segmentation AUPRO: 87.4
Segmentation AUPRO (until 30% FPR): 87.4
defect-detection-on-kolektorsdd2SuperSimpleNet
Average Precision: 97.4

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SuperSimpleNet:面向快速可靠表面缺陷检测的无监督与有监督学习统一方法 | 论文 | HyperAI超神经