File size: 1,377 Bytes
4479f79
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
export config_path="./train/config/XVerse_config_INF.yaml"
export model_checkpoint="./checkpoints/XVerse"
export target_size=768
export condition_size=256
export test_list_name="XVerseBench_single"
export save_name="./eval/XVerseBench_singleidip"

ports=(`echo $METIS_WORKER_0_PORT | tr ',' ' '`)
port=${ports[-1]}

accelerate launch \
    --main_process_port $port \
    -m eval.tools.idip_gen_split_idip \
    --config_name "$config_path" \
    --model_path "$model_checkpoint" \
    --target_size "$target_size" \
    --condition_size "$condition_size" \
    --save_name "$save_name" \
    --test_list_name "$test_list_name"

accelerate launch \
    --main_process_port $port \
    -m eval.tools.idip_dpg_score \
    --input_dir "$save_name" \
    --test_list_name "$test_list_name"

accelerate launch \
    --main_process_port $port \
    -m eval.tools.idip_aes_score \
    --input_dir "$save_name" \
    --test_list_name "$test_list_name"

accelerate launch \
    --main_process_port $port \
    -m eval.tools.idip_face_score \
    --input_dir "$save_name" \
    --test_list_name "$test_list_name"

accelerate launch \
    --main_process_port $port \
    -m eval.tools.idip_sam-dino_score \
    --input_dir "$save_name" \
    --test_list_name "$test_list_name"

python \
    -m eval.tools.log_scores \
    --input_dir "$save_name" \
    --test_list_name "$test_list_name"