README.md

Download Datasets

• MVTec-LOCO [Official]
• MVTec-AD [Official]
• VisA [Official]
• Real-IAD [Official]

Place them in this folder. MVTec-AD and MVTec-LOCO do not require preprocessing. For VisA, please use the following command for preprocessing:

$ python prepare_visa.py --data_folder <Downloaded-VisA-Dataset> --save_folder visa  --split_file split_csv/1cls.csv

For Real-IAD, use the following command:

$ python prepare_realiad.py --data_folder <Downloaded-Real-IAD-Dataset> --save_folder realiad  --split_file_root realiad_jsons_sv

Download Segmentation Maps

• mvtec_loco segmentations
• mvtec segmentations
• visa segmentations
• real-iad segmentations

Directory Structure

  |--data                       
      |--mvtec
          |--bottle
            |--ground_truth
            |--test
            |--train
          |--cable
          |--...
      |--visa
          |--candle
            |--ground_truth
            |--test
            |--train
          |--capsules
          |--...
      |--mvtec_loco
          |--breakfast_box
            |--ground_truth
            |--test
            |--train
            |--validation
            |--defects_config.json
          |--juice_bottle
          |--...
      |--realiad
          |--audiojack
            |--ground_truth
            |--test
            |--train
          |--bottle_cap
          |--...
      |--mvtec_segmentations
          |--bottle
            |--test
            |--train
          |--cable
          |--...
      |--visa_segmentations
      |--mvtec_loco_segmentations
      |--realiad_segmentations