-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathrun.sh
69 lines (49 loc) · 1.69 KB
/
run.sh
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
# generation args
save_dir="outputs"
prompt="Sandy beach, large driftwood in the foreground, calm sea beyond, realism style."
scene_type="outdoor"
lora_path='checkpoints/pano_lora_720*1440_v1.safetensors'
depth_model='DepthAnythingv2'
seed=16806
# [step-1]: generate reference panorama
python gen_refpano.py \
--save_dir ${save_dir} \
--prompt "${prompt}" \
--lora_path ${lora_path} \
--seed ${seed} \
# [step-2]: autolayering
python gen_panodepth.py \
--save_dir ${save_dir} \
--depth_model ${depth_model} \
--input_path "${save_dir}/rgb.png" \
python gen_autolayering.py \
--input_dir ${save_dir} \
--scene_type ${scene_type} \
# [step-3]: construct layered RGB panoramas
python gen_layerdata.py \
--lora_path ${lora_path} \
--base_dir "${save_dir}/layering" \
--seed ${seed} \
# [optional SR step]
python pasd/run_layers_pasd.py \
--inputs_dir "${save_dir}/layering" \
--upscale 2
# [step-3]: construct layered 3D panoramas(rgb+depth)
python gen_traindata.py \
--save_dir ${save_dir} \
--depth_model ${depth_model} \
--layerpano_dir "${save_dir}/layering" \
--sr # use sr panoramas(if you run the above SR step)
# [step-4]: Panoramic 3D Gaussian Scene construction
python run_layerpano.py \
--input_dir ${save_dir} \
--save_dir ${save_dir}
## Rendering args
elevation=0
## [Rendering Visualization]
python -m rendering.render_video_360 \
--save_dir "${save_dir}/scene" \
--elevation ${elevation}
python -m rendering.render_video_zigzag \
--save_dir "${save_dir}/scene"
# apptainer exec --nv --bind /mnt:/mnt /mnt/petrelfs/yangshuai/apptainer/panodepth.sif bash run.sh