# 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