MvtecAD unsupervised Anomaly Detection

Last update: Feb 25, 2022

Related tags

Overview

MvtecAD unsupervised Anomaly Detection

This respository is the unofficial implementations of DFR: Deep Feature Reconstruction for Unsupervised Anomaly Segmentation

Result of 500 epochs trained model

Selects latent sizes of Autoencoder by PCA

Classes	latent size	Segmentation AUC	Detection AUC
bottle	116	97.2771%	99.8413%
cable	557	95.5101%	84.8951%
capsule	162	98.8928%	97.3275%
carpet	245	97.9116%	90.5297%
grid	145	97.2484	79.5322%
hazelnut	459	98.5848%	100%
leather	325	98.8649%	95.4484%
metal_nut	380	96.127%	97.263%
pill	292	98.0543%	94.108%
screw	283	99.3001%	92.0066%
tile	557	89.4887%	91.7388%
toothbrush	243	98.6729%	91.3889%
transistor	333	83.9157%	89.0833%
wood	364	91.7027%	98.9474%
zipper	115	95.6663%	83.2983%
mean		95.8141%	92.3606

How to run

requirements

pytorch scikit-learn matplotlib numpy pandas PIL wget

Train

python main.py --mode          train      
               --data_dir_head [Datapath] 
               --BATCH_SIZE    [BATCH_SIZE] 
               --LR            [Learning Rate] 
               --EPOCH         [Epochs] 
               --backbone      [Feature map of Conv in VGG19]
               --latent_dim    [Latent size of CAE] 
               --classes       [Default is all]

Download 500 Epochs Finetuned Models

Here provide the model of each classes in Drophox

python main.py --mode download

Evaluate the ROC-AUC of Test Set

python main.py --mode        evaluation    
               --classes     [Default is all]

Inference the model

python main.py --mode           inference    
               --heatmap_path   [Input path] 
               --heatmap_item   [Class of input] 
               --heatmap_gt     [GT path Default is None]
               --device         [cpu or cuda]
               --device         [Output path ]

Example run in main.py

if __name__ == "__main__":  
    cfg = config()
    cfg.mode = "inference"
    cfg.heatmap_path = 'mvtecad_unsupervise/bottle/test/broken_small/001.png'
    cfg.heatmap_item = 'bottle'
    cfg.heatmap_gt = 'mvtecad_unsupervise/bottle/ground_truth/broken_small/001_mask.png'
    cfg.device = 'cpu'
    cfg.heatmap_export = 'validate/Inferece.png'

validate/Inferece.png is

Code Reference

https://github.com/YoungGod/DFR

https://www.kaggle.com/danieldelro/unsupervised-anomaly-segmentation-of-screw-images

MvtecAD unsupervised Anomaly Detection

Related tags

Overview

MvtecAD unsupervised Anomaly Detection

Result of 500 epochs trained model

Selects latent sizes of Autoencoder by PCA

How to run

requirements

Train

Download 500 Epochs Finetuned Models

Evaluate the ROC-AUC of Test Set

Inference the model

Code Reference

Owner

PyTorch implementation for COMPLETER: Incomplete Multi-view Clustering via Contrastive Prediction (CVPR 2021)

An updated version of virtual model making

OneFlow is a performance-centered and open-source deep learning framework.

Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561

Scripts used to make and evaluate OpenAlex's concept tagging model

DeepFaceLab fork which provides IPython Notebook to use DFL with Google Colab

Code base for the paper "Scalable One-Pass Optimisation of High-Dimensional Weight-Update Hyperparameters by Implicit Differentiation"

This repository contains the official implementation code of the paper Transformer-based Feature Reconstruction Network for Robust Multimodal Sentiment Analysis

Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization.

Calibrated Hyperspectral Image Reconstruction via Graph-based Self-Tuning Network.

Qimera: Data-free Quantization with Synthetic Boundary Supporting Samples

Alphabetical Letter Recognition

SOTR: Segmenting Objects with Transformers [ICCV 2021]

Author Disambiguation using Knowledge Graph Embeddings with Literals

This is an example of object detection on Micro bacterium tuberculosis using Mask-RCNN

A Keras implementation of CapsNet in the paper: Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules

Digan - Official PyTorch implementation of Generating Videos with Dynamics-aware Implicit Generative Adversarial Networks

This is the source code for generating the ASL-Skeleton3D and ASL-Phono datasets. Check out the README.md for more details.

Source code for paper "Document-Level Relation Extraction with Adaptive Thresholding and Localized Context Pooling", AAAI 2021

Sudoku solver - A sudoku solver with python