HolisticDiffuScene/run_3dfront_room_layout_sample.sh at develop · fangchuan/HolisticDiffuScene · GitHub

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
# bedroom parametrization
# MODEL_FLAGS="--layout_channels 33 --layout_size 23 --num_channels 128 --num_res_blocks 3 --b_learn_sigma True --b_class_cond True --b_text_cond False --use_input_encoding False"

# living room parametrization
# MODEL_FLAGS="--layout_channels 36 --layout_size 41 --num_channels 128 --num_res_blocks 3 --b_learn_sigma True --b_class_cond False --b_text_cond True --use_input_encoding False"

# DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine  --timestep_respacing 250"


# python scripts/threed_front_room_layout_sample.py --data_dir /mnt/nas_3dv/hdd1/datasets/3D_FRONT_FUTURE/full/test/livingroom/ \
#     --model_path log/openai-2023-09-23-20-13-33-964075/ema_0.9999_100000.pt \
#      $MODEL_FLAGS \
#      $DIFFUSION_FLAGS \
#      --num_samples 1000 \
#       --room_type 'livingroom' \
#       --path_to_pickled_3d_futute_models /mnt/nas_3dv/hdd1/datasets/3D_FRONT_FUTURE/threed_future_model_livingroom.pkl


# MODEL_FLAGS="--layout_channels 30 --layout_size 13 --num_channels 128 --num_res_blocks 3 --b_learn_sigma True --b_class_cond False --b_text_cond True --use_input_encoding True"

# DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine  --timestep_respacing ddim200 --use_ddim True"
# CONFIG_FILE="/mnt/nas_3dv/hdd1/fangchuan/HolisticDiffuScene/config/3dfront_bedroom_config.yaml"

# python scripts/threed_front_room_layout_sample2.py --data_dir /mnt/nas_3dv/hdd1/datasets/3D_FRONT_FUTURE/holistic_layout_20231113/threed_front_bedroom/ \
#     --model_path log/3dfront_bedroom/openai-2023-11-13-11-03-22-754137/ema_0.9999_100000.pt \
#      $MODEL_FLAGS \
#      $DIFFUSION_FLAGS \
#      --num_samples 1000 \
#       --room_type 'bedroom' \
#       --path_to_pickled_3d_futute_models /mnt/nas_3dv/hdd1/datasets/3D_FRONT_FUTURE/threed_future_model_bedroom.pkl \
#       --config_file $CONFIG_FILE \

MODEL_FLAGS="--layout_channels 33 --layout_size 21 --num_channels 128 --num_res_blocks 3 --b_learn_sigma True --b_class_cond False --b_text_cond True --use_input_encoding True"
DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine  --timestep_respacing ddim200 --use_ddim True"
CONFIG_FILE="/mnt/nas_3dv/hdd1/fangchuan/HolisticDiffuScene/config/3dfront_livingroom_config.yaml"

python scripts/threed_front_room_layout_sample2.py \
    --model_path log/3dfront_livingroom/openai-2023-11-15-19-59-23-819741/ema_0.9999_200000.pt \
     $MODEL_FLAGS \
     $DIFFUSION_FLAGS \
     --num_samples 1000 \
      --room_type 'livingroom' \
      --path_to_pickled_3d_futute_models /mnt/nas_3dv/hdd1/datasets/3D_FRONT_FUTURE/threed_future_model_livingroom.pkl \
      --config_file $CONFIG_FILE \
      --log_dir "log/3dfront_livingroom/text-cond-train-test"