Hugging Face's logo

diing
/
AURAD

Model card Files
xet
 Community
AURAD
File size: 3,413 Bytes
41c8683
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# nohup env CUDA_VISIBLE_DEVICES=1 python generate_layout.py \
#     --model_dir /home/AURA/text2layout/instruct-pix2pix-model/ \
#     --checkpoint_num 7000 \
#     --image_root  /home/AURA/dataset \
#     --test_json_path /home/AURA/dataset/json/text2layout/test_prompt_text2layout_single.json \
#     --generated_num 1


# CUDA_VISIBLE_DEVICES=1  python generate_layout_rib.py \
#   --output_dir /data16T/chestx-ray/AURAD_infer/det-test-syn-mask-cp10000 \
#   --model_dir /home/shuhan/blobdata_sd/AURA/text2layout/total \
#   --checkpoint_num 10000 \
#   --image_root /data16T/chestx-ray/AURAD_dataset \
#   --test_json_path /data16T/chestx-ray/AURAD_dataset/chexlocalize/json-layout2image/test_prompt_text2layout_single.json \
#   --max_attempts 5


# python generate_layout_rib.py \
#   --output_dir /data16T/chestx-ray/AURAD_infer/det-test-syn-mask-cp10000 \
#   --model_dir /home/shuhan/blobdata_sd/AURA/text2layout/total \
#   --checkpoint_num 10000 \
#   --image_root /data16T/chestx-ray/AURAD_dataset \
#   --test_json_path /data16T/chestx-ray/AURAD_dataset/chexlocalize/json-layout2image/test_prompt_text2layout_single.json \
#   --max_attempts 5


python generate_layout_rib.py \
  --output_dir /home/jovyan/AURAD_infer/mask/det-test-cp10000 \
  --model_dir /home/jovyan/AURAD/text2layout/total_mask01 \
  --checkpoint_num 9000 \
  --pretrained_model_name_or_path CompVis/stable-diffusion-v1-4 \
  --image_root /home/jovyan/AURAD_dataset \
  --test_json_path /home/jovyan/AURAD_dataset/chestdet/test_prompt_text2layout_single.json \
  --resolution 512 \
  --max_attempts 5 \
  --num_inference_steps 50 \
  --guidance_scale 7.0 \
  --image_guidance_scale 1.5 \
  --mask_threshold 0.3 \
  --morph_radius 5 \
  --morph_mode open \
  --seed 42 \
  # --save_full_rgb


python generate_layout_rib.py \
  --output_dir /home/jovyan/AURAD_infer/mask/det-test-cp9000-2 \
  --model_dir /home/jovyan/AURAD/text2layout/total_mask01 \
  --checkpoint_num 9000 \
  --pretrained_model_name_or_path CompVis/stable-diffusion-v1-4 \
  --image_root /home/jovyan/AURAD_dataset \
  --test_json_path /home/jovyan/AURAD_dataset/chestdet/test_prompt_text2layout_single.json \
  --resolution 512 \
  --max_attempts 5 \
  --num_inference_steps 50 \
  --guidance_scale 9.0 \
  --image_guidance_scale 1 \
  --mask_threshold 0.3 \
  --morph_radius 5 \
  --morph_mode open \
  --seed 42 


python generate_layout_rib.py \
  --output_dir /home/jovyan/AURAD_infer/mask/mimic-followup-cp10000 \
  --model_dir /home/jovyan/AURAD/text2layout/total_mask01 \
  --checkpoint_num 9000 \
  --pretrained_model_name_or_path CompVis/stable-diffusion-v1-4 \
  --image_root /home/jovyan/AURAD_dataset/mimic-followup \
  --test_json_path /home/jovyan/AURAD_dataset/mimic-followup/train_100_specific5x_single.json \
  --resolution 512 \
  --max_attempts 5 \
  --num_inference_steps 50 \
  --guidance_scale 9.0 \
  --image_guidance_scale 1 \
  --mask_threshold 0.3 \
  --morph_radius 5 \
  --morph_mode open \
  --seed 42 



python generate_layout_no_filter.py \
  --output_dir /home/jovyan/AURAD_infer/mask/det-test-cp10000 \
  --model_dir /home/jovyan/AURAD/text2layout/total_mask01 \
  --checkpoint_num 9000 \
  --pretrained_model_name_or_path CompVis/stable-diffusion-v1-4 \
  --image_root /home/jovyan/AURAD_dataset \
  --test_json_path /home/jovyan/AURAD_dataset/chestdet/test_prompt_text2layout_single.json \
  --seed 42 \
  --save_full_rgb