search

tiskw / patchcore-ad

Unofficial implementation of PatchCore and several additional experiments.
MIT License
12 stars 1 forks source link

New function: contour map visualization #7

Closed tiskw closed 4 months ago

tiskw commented 4 months ago

Add code for contour map visualization.

The following is the usage example for the capsule dataset of MVTecAD dataset.

# Train.
python3 main.py train -i data/mvtec_ad/capsule/train/good -o ./index.faiss

# Compute threshold.
python3 main.py thresh -i data/mvtec_ad/capsule/train/good

# Visualize contour map using the threshold value obtained by the above.
python3 main.py predict --contour 1.47 -i data/mvtec_ad/capsule/test/crack          -o output_test/crack
python3 main.py predict --contour 1.47 -i data/mvtec_ad/capsule/test/faulty_imprint -o output_test/faulty_imprint
python3 main.py predict --contour 1.47 -i data/mvtec_ad/capsule/test/good           -o output_test/good
python3 main.py predict --contour 1.47 -i data/mvtec_ad/capsule/test/poke           -o output_test/poke
python3 main.py predict --contour 1.47 -i data/mvtec_ad/capsule/test/scratch        -o output_test/scratch
python3 main.py predict --contour 1.47 -i data/mvtec_ad/capsule/test/squeeze        -o output_test/squeeze