-
Notifications
You must be signed in to change notification settings - Fork 80
Expand file tree
/
Copy pathrun_KSDD2.py
More file actions
29 lines (21 loc) · 760 Bytes
/
run_KSDD2.py
File metadata and controls
29 lines (21 loc) · 760 Bytes
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
import os
from datasets import dataset_classes
from multiprocessing import Pool
if __name__ == '__main__':
pool = Pool(processes=1)
dataset_list = ['ksdd2']
gpu_indx = 0
for dataset in dataset_list:
classes = dataset_classes[dataset]
for cls in classes[:]:
sh_method = f'python eval_SAA.py ' \
f'--dataset {dataset} ' \
f'--class-name {cls} ' \
f'--batch-size {1} ' \
f'--root-dir ./result_KSDD2 ' \
f'--cal-pro False ' \
f'--gpu-id {gpu_indx} ' \
print(sh_method)
pool.apply_async(os.system, (sh_method,))
pool.close()
pool.join()