Skip to content

[TF 2.x] PaDiM - unofficial tensorflow implementation of the paper 'a Patch Distribution Modeling Framework for Anomaly Detection and Localization'.

Notifications You must be signed in to change notification settings

remmarp/PaDiM-TF

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

[TF 2.x] PaDiM - Anomaly Detection Localization

This repository contains an unofficial PaDiM implementation using tensorflow.

Paper

PaDiM: a patch distribution modeling framework for anomaly detection and localization. [Link]

Dependencies

  • Windows 10, Python 3.8.8, Tensorflow 2.4.1 GPU
  • Scikit-learn, Scikit-image, Matplotlib

Run

# options: seed, rd, target, batch_size, is_plot, net
python main.py

Dataset

MVTecAD dataset

Results (AU ROC)

Implementation results on MVTec

  • Network Type:

    • PyTorch.: WideResNet50, Rd 550 (from PyTorch version) (WR50-Rd550)
    • Net 1: EfficientNetB7 [layer a_expand_activation 5, 6, 7], Rd 1000 (ENB7-Rd1000)
    • Net 2: EfficientNetB7 [layer a_expand_activation 4, 6, 7], Rd 1000 (ENB7-Rd1000)
    • Net 3: EfficientNetB7 [layer a_activation 5, 6, 7], Rd 1000 (ENB7-Rd1000)
  • I observed that intermediate layers selection has some effects on detection performance.

  • Besides, a high image-level au-roc does not guarantee a high level of au-roc on patch-level.

MvTec PyTorch (Img) Net 1 (Img) Net 2 (Img) Net 3 (Img)
carpet 0.999 0.950 0.982 0.996
grid 0.957 0.936 0.971 0.976
leather 1.000 0.999 1.000 1.000
tile 0.974 0.957 0.984 0.981
wood 0.988 0.948 0.954 0.990
bottle 0.998 0.983 0.996 0.999
cable 0.922 0.909 0.919 0.973
capsule 0.915 0.946 0.953 0.958
hazelnut 0.933 0.983 0.973 0.997
metal_nut 0.992 0.869 0.930 0.931
pill 0.944 0.882 0.879 0.925
screw 0.844 0.632 0.767 0.895
toothbrush 0.972 0.767 0.972 0.811
transistor 0.978 0.930 0.949 0.975
zipper 0.909 0.980 0.986 0.990
Avg. (tex.) 0.9840 0.9579 0.9781 0.9885
Avg. (obj.) 0.9410 0.8881 0.9323 0.9455
Avg. (all) 0.9550 0.9114 0.9476 0.9598
MvTec org. (Patch) Net 1 (Patch) Net 2 (Patch) Net 3(Patch)
carpet 0.990 0.973 0.854 0.829
grid 0.965 0.958 0.750 0.768
leather 0.989 0.986 0.902 0.831
tile 0.939 0.905 0.729 0.748
wood 0.941 0.946 0.831 0.814
bottle 0.982 0.971 0.861 0.831
cable 0.968 0.963 0.815 0.843
capsule 0.986 0.977 0.940 0.911
hazelnut 0.979 0.965 0.876 0.834
metal_nut 0.971 0.986 0.926 0.926
pill 0.961 0.955 0.893 0.903
screw 0.983 0.986 0.941 0.893
toothbrush 0.983 0.979 0.937 0.864
transistor 0.987 0.977 0.958 0.958
zipper 0.975 0.965 0.840 0.814
Avg. (tex.) 0.9650 0.9536 0.8131 0.7979
Avg. (obj.) 0.9780 0.9724 0.8987 0.8776
Avg. (all) 0.9730 0.9661 0.8702 0.8510

ROC Curve (Net 1) Bottle

bottle_auroc

PR Curve (Net 1) Bottle

bottle_pr

Localization examples (Net 1) (cherry-picked)

carpet_ex grid_ex leather_ex tile_ex wood_ex bottle_ex cable_ex capsule_ex hazelnut_ex metalnut_ex pill_ex screw_ex toothbrush_ex transistor_ex zipper_ex

About

[TF 2.x] PaDiM - unofficial tensorflow implementation of the paper 'a Patch Distribution Modeling Framework for Anomaly Detection and Localization'.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages