conda create -n fmad python==3.10 conda activate fmad

torch >= 1.12.1

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

pip install -r requirements.txt

Training

python main.py --data_path '$Main folder to MVTec or VisA$' --dataset mvtec # 'mvtec' or 'visa', for eva on visa, please use 'create_visa_dataset.py' to create mvtec style dataset

Testing

python infer.py --class_name bottle