MetaUAS/test.sh at main · gaobb/MetaUAS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

# metauas on mvtec with 256x256 inputs
model_name='weights/metauas-256.ckpt'
test_json='./data/MVTec-AD/test.json'
image_dir='../../datasets/mvtec/'
for seed in 1 3 5 7 9
do
      prompt_json="./data/MVTec-AD/oneprompt_seed${seed}.json"
      CUDA_VISIBLE_DEVICES="0" python3 test.py \
            --checkpoint $model_name  \
            --seed $seed \
            --img_size 256 \
            --image_dir $image_dir \
            --test_json $test_json \
            --prompt_json $prompt_json
done
# metauas-star on mvtec with 256x256 inputs
prompt_json='./data/MVTec-AD/test-train-top10pair-eb4.json'
CUDA_VISIBLE_DEVICES="0" python3 test.py \
            --checkpoint $model_name  \
            --seed $seed \
            --img_size 256 \
            --image_dir $image_dir \
            --test_json $test_json \
            --prompt_json $prompt_json


# metauas on visa with 512x512 inputs
model_name='weights/metauas-512.ckpt'
test_json='./data/VisA-AD/test.json'
image_dir='../../datasets/visa/'
for seed in 1 3 5 7 9
do
      prompt_json="./data/VisA-AD/oneprompt_seed${seed}.json"
      CUDA_VISIBLE_DEVICES="0" python3 test.py \
            --checkpoint $model_name  \
            --seed $seed \
            --img_size 512 \
            --image_dir $image_dir \
            --test_json $test_json \
            --prompt_json $prompt_json
done
# metauas-star on visa with 512x512 inputs
prompt_json='./data/VisA-AD/test-train-top10pair-eb4.json'
CUDA_VISIBLE_DEVICES="0" python3 test.py \
            --checkpoint $model_name  \
            --seed $seed \
            --img_size 512 \
            --image_dir $image_dir \
            --test_json $test_json \
            --prompt_json $prompt_json


# metauas on goodsad with 512x512 inputs
model_name='weights/metauas-512.ckpt'
test_json='./data/GoodsAD/test.json'
image_dir='/home/tione/notebook/datasets/LSAD/GoodsAD/'
for seed in 1 3 5 7 9
do
      prompt_json="./data/GoodsAD/oneprompt_seed${seed}.json"
      CUDA_VISIBLE_DEVICES="0" python3 test.py \
            --checkpoint $model_name  \
            --seed $seed \
            --img_size 512 \
            --image_dir $image_dir \
            --test_json $test_json \
            --prompt_json $prompt_json
done
# metauas-star on goodsad with 512x512 inputs
prompt_json='./data/GoodsAD/test-train-top10pair-eb4.json'
CUDA_VISIBLE_DEVICES="0" python3 test.py \
            --checkpoint $model_name  \
            --seed $seed \
            --img_size 512 \
            --image_dir $image_dir \
            --test_json $test_json \
            --prompt_json $prompt_json