boringKey/MSTAD

★ 2Forks 0PythonGitHub ↗Compare

README

MSTAD: a Masked Subspace-like Transformer for multi-class Anomaly Detection

This is the code implementation for the paper titled: "MSTAD: a Masked Subspace-like Transformer for multi-class Anomaly Detection" (Accepted to Knowledge-based Systems)

Requirements

einops==0.4.1
numpy==1.19.5
opencv-python==3.4.10.37
Pillow==8.3.2
PyYAML==5.3.1
scikit-learn==0.24.2
scipy==1.5.4
tabulate==0.8.10
tensorboardX==1.8
torch==1.5.0
torchvision==0.6.0+cu90

Contributors

boringKey

Issues