PatchCore anomaly detection

Unofficial implementation of PatchCore(new SOTA) anomaly detection model

Original Paper : Towards Total Recall in Industrial Anomaly Detection (Jun 2021)
Karsten Roth, Latha Pemula, Joaquin Zepeda, Bernhard Schölkopf, Thomas Brox, Peter Gehler

https://arxiv.org/abs/2106.08265
https://paperswithcode.com/sota/anomaly-detection-on-mvtec-ad

notice(21/06/18) :
This code is not yet verified. Any feedback is appreciated.
updates(21/06/21) :

I used sklearn's SparseRandomProjection(ep=0.9) for random projection. I'm not confident with this.
I think exact value of "b nearest patch-features" is not presented in the paper. I just set 9. (args.n_neighbors)
In terms of NN search, author used "faiss". but not implemented in this code yet.
sample embeddings/carpet/embedding.pickle => coreset_sampling_ratio=0.001

updates(21/06/26) :

A critical issue related to "locally aware patch" raised and fixed. Score table is updated.

Usage

# install python 3.6, torch==1.8.1, torchvision==0.9.1
pip install -r requirements.txt

python train.py --phase train or test --dataset_path .../mvtec_anomaly_detection --category carpet --project_root_path path/to/save/results --coreset_sampling_ratio 0.01 --n_neighbors 9'

# for fast try just specify your dataset_path and run
python train.py --phase test --dataset_path .../mvtec_anomaly_detection --project_root_path ./

MVTecAD AUROC score (PatchCore-1%, mean of n trials)

Category	Paper (image-level)	This code (image-level)	Paper (pixel-level)	This code (pixel-level)
carpet	0.980	0.991(1)	0.989	0.989(1)
grid	0.986	0.975(1)	0.986	0.975(1)
leather	1.000	1.000(1)	0.993	0.991(1)
tile	0.994	0.994(1)	0.961	0.949(1)
wood	0.992	0.989(1)	0.951	0.936(1)
bottle	1.000	1.000(1)	0.985	0.981(1)
cable	0.993	0.995(1)	0.982	0.983(1)
capsule	0.980	0.976(1)	0.988	0.989(1)
hazelnut	1.000	1.000(1)	0.986	0.985(1)
metal nut	0.997	0.999(1)	0.984	0.984(1)
pill	0.970	0.959(1)	0.971	0.977(1)
screw	0.964	0.949(1)	0.992	0.977(1)
toothbrush	1.000	1.000(1)	0.985	0.986(1)
transistor	0.999	1.000(1)	0.949	0.972(1)
zipper	0.992	0.995(1)	0.988	0.984(1)
mean	0.990	0.988	0.980	0.977

Code Reference

kcenter algorithm :
https://github.com/google/active-learning
embedding concat function :
https://github.com/xiahaifeng1995/PaDiM-Anomaly-Detection-Localization-master

Unofficial implementation of PatchCore anomaly detection

Related tags

Overview

PatchCore anomaly detection

Usage

MVTecAD AUROC score (PatchCore-1%, mean of n trials)

Code Reference

Owner

Changwoo Ha

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO

SARS-Cov-2 Recombinant Finder for fasta sequences

Deep learning PyTorch library for time series forecasting, classification, and anomaly detection

Official repository of the paper "A Variational Approximation for Analyzing the Dynamics of Panel Data". Mixed Effect Neural ODE. UAI 2021.

An example project demonstrating how the Autonomous Learning Library can be used to build new reinforcement learning agents.

Rest API Written In Python To Classify NSFW Images.

PrimitiveNet: Primitive Instance Segmentation with Local Primitive Embedding under Adversarial Metric (ICCV 2021)

A disassembler for the RP2040 Programmable I/O State-machine!

Noise Conditional Score Networks (NeurIPS 2019, Oral)

Official implementation of the ICCV 2021 paper "Conditional DETR for Fast Training Convergence".

Using knowledge-informed machine learning on the PRONOSTIA (FEMTO) and IMS bearing data sets. Predict remaining-useful-life (RUL).

Code for “ACE-HGNN: Adaptive Curvature ExplorationHyperbolic Graph Neural Network”

FairMOT for Multi-Class MOT using YOLOX as Detector

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.

QSYM: A Practical Concolic Execution Engine Tailored for Hybrid Fuzzing

Official implementation of Sparse Transformer-based Action Recognition

3D ResNets for Action Recognition (CVPR 2018)

TensorFlow-LiveLessons - "Deep Learning with TensorFlow" LiveLessons

Transfer Learning Remote Sensing

Paper: De-rendering Stylized Texts