Transformer-Based Contractive Learning Framework for Anomaly Detection

source code of Transformer-Based Contractive Learning Framework for Anomaly Detection

Requirements

pyhton >= 3.7.3
torch >= 1.9.0
torchvision >= 0.10.0
pytorch_pretrained_vit >=0.0.7

Training on CIFAR10

To train the model on CIFAR10 dataset for a given normal class, run the following:

python main.py