-
Notifications
You must be signed in to change notification settings - Fork 2
/
demo_run_batch_replica.py
59 lines (46 loc) · 3.86 KB
/
demo_run_batch_replica.py
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
import subprocess
import sys
data='/mnt/ws-frb/projects/gaussian_splatting/data_for_original_gs/replica/office0'
output_path = '/mnt/ws-frb/users/frank/frank/gaussian_splatting/'
iter='2000'
folder_path='demo_replica_office0_id79/voxel_size_0.1/'
frame_ids = '79'
voxel_size=0.1
suffix=''
cuda_device='0'
save_iterations = '1 500 1000 2000 2200'
opacity_reset_interval='1000' # default: 3000
# Baselines
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/only_c --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval}"
subprocess.run(command+suffix, shell=True)
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/only_ds --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --DS 0.1 --CS 0"
subprocess.run(command+suffix, shell=True)
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/ds_fovmask --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --DS 0.1 --fov_mask"
subprocess.run(command+suffix, shell=True)
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/dist_loss --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --dist"
subprocess.run(command+suffix, shell=True)
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/reset_opa --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --reset_opa_near"
subprocess.run(command+suffix, shell=True)
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/reset_opa_fovmask --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --reset_opa_near --fov_mask"
subprocess.run(command+suffix, shell=True)
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/ds_reset_opa_fovmask --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --DS 0.1 --reset_opa_near --fov_mask"
subprocess.run(command+suffix, shell=True)
# Ours
command = f"CUDA_VISIBLE_DEVICES={cuda_device} python train.py -s {data}\
-m {output_path}{folder_path}/SAD_GS --single_frame_id {frame_ids}\
--save_iterations {save_iterations} --iterations {iter} --checkpoint_iterations {iter} --init_w_gaussian --voxel_size {voxel_size} --densify_from_iter 100 --opacity_reset_interval {opacity_reset_interval} --CS 10 --cls_loss --full_reset_opa"
subprocess.run(command+suffix, shell=True)