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

本步态识别系统主要基于GaitSet模型进行实现

Gapmm2: gapped alignment using minimap2 (align transcripts to genome)

Air Pollution Prediction System using Linear Regression and ANN

PyTorch implementation for 3D human pose estimation

ivadomed is an integrated framework for medical image analysis with deep learning.

Open-source implementation of Google Vizier for hyper parameters tuning

Minimal PyTorch implementation of Generative Latent Optimization from the paper "Optimizing the Latent Space of Generative Networks"

Apache Spark - A unified analytics engine for large-scale data processing

Model-free Vehicle Tracking and State Estimation in Point Cloud Sequences

An official implementation of "SFNet: Learning Object-aware Semantic Correspondence" (CVPR 2019, TPAMI 2020) in PyTorch.

SAS: Self-Augmentation Strategy for Language Model Pre-training

Codes for our IJCAI21 paper: Dialogue Discourse-Aware Graph Model and Data Augmentation for Meeting Summarization

Detecting and Tracking Small and Dense Moving Objects in Satellite Videos: A Benchmark

Implementation of "The Power of Scale for Parameter-Efficient Prompt Tuning"

A basic duplicate image detection service using perceptual image hash functions and nearest neighbor search, implemented using faiss, fastapi, and imagehash

QT Py Media Knob using rotary encoder & neopixel ring

Rayvens makes it possible for data scientists to access hundreds of data services within Ray with little effort.

[ICCV'21] Neural Radiance Flow for 4D View Synthesis and Video Processing

AutoVideo: An Automated Video Action Recognition System

This repo implements a 3D segmentation task for an airport baggage dataset.