anomaly-detection
-
Anomaly detection
- vanilla CNN (supervised)
- resnet18 (supervised)
-
Dependencies
- Python 3.6+
- PyTorch==1.3
- Dataset: MVTec Anomaly Detection Dataset
Dataset structure
./data
│
├── bottle
│ ├── ground_truth
│ ├── train
│ └── test
│
├── carpet
├── leather
├── grid
Train model
- Run the following command.
python train.py --target bottle --model resnet18 -c 0 --lr 0.001