Hugging Face's logo

diing
/
AURAD

Model card Files
xet
 Community
AURAD / text2layout /generate_layout.sh
diing's picture
diing
Upload folder using huggingface_hub
41c8683 verified
Raw History Blame Contribute Delete
3.41 kB
# 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