divya-22 commited on
Commit
63a066c
1 Parent(s): 28d3e59

Upload 2 files

Browse files
Files changed (2) hide show
  1. configs/image.yaml +69 -0
  2. configs/text.yaml +68 -0
configs/image.yaml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ### Input
2
+ # input rgba image path (default to None, can be load in GUI too)
3
+ input:
4
+ # input text prompt (default to None, can be input in GUI too)
5
+ prompt:
6
+ # input mesh for stage 2 (auto-search from stage 1 output path if None)
7
+ mesh:
8
+ # estimated elevation angle for input image
9
+ elevation: 0
10
+ # reference image resolution
11
+ ref_size: 256
12
+ # density thresh for mesh extraction
13
+ density_thresh: 1
14
+
15
+ ### Output
16
+ outdir: logs
17
+ mesh_format: obj
18
+ save_path: ???
19
+
20
+ ### Training
21
+ # guidance loss weights (0 to disable)
22
+ lambda_sd: 0
23
+ lambda_zero123: 1
24
+ # training batch size per iter
25
+ batch_size: 1
26
+ # training iterations for stage 1
27
+ iters: 500
28
+ # training iterations for stage 2
29
+ iters_refine: 50
30
+ # training camera radius
31
+ radius: 2
32
+ # training camera fovy
33
+ fovy: 49.1 # align with zero123 rendering setting (ref: https://github.com/cvlab-columbia/zero123/blob/main/objaverse-rendering/scripts/blender_script.py#L61
34
+ # checkpoint to load for stage 1 (should be a ply file)
35
+ load:
36
+ # whether allow geom training in stage 2
37
+ train_geo: False
38
+ # prob to invert background color during training (0 = always black, 1 = always white)
39
+ invert_bg_prob: 0.5
40
+
41
+
42
+ ### GUI
43
+ gui: False
44
+ force_cuda_rast: False
45
+ # GUI resolution
46
+ H: 800
47
+ W: 800
48
+
49
+ ### Gaussian splatting
50
+ num_pts: 5000
51
+ sh_degree: 0
52
+ position_lr_init: 0.001
53
+ position_lr_final: 0.00002
54
+ position_lr_delay_mult: 0.02
55
+ position_lr_max_steps: 500
56
+ feature_lr: 0.01
57
+ opacity_lr: 0.05
58
+ scaling_lr: 0.005
59
+ rotation_lr: 0.005
60
+ percent_dense: 0.1
61
+ density_start_iter: 100
62
+ density_end_iter: 3000
63
+ densification_interval: 100
64
+ opacity_reset_interval: 700
65
+ densify_grad_threshold: 0.5
66
+
67
+ ### Textured Mesh
68
+ geom_lr: 0.0001
69
+ texture_lr: 0.2
configs/text.yaml ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ### Input
2
+ # input rgba image path (default to None, can be load in GUI too)
3
+ input:
4
+ # input text prompt (default to None, can be input in GUI too)
5
+ prompt:
6
+ # input mesh for stage 2 (auto-search from stage 1 output path if None)
7
+ mesh:
8
+ # estimated elevation angle for input image
9
+ elevation: 0
10
+ # reference image resolution
11
+ ref_size: 256
12
+ # density thresh for mesh extraction
13
+ density_thresh: 1
14
+
15
+ ### Output
16
+ outdir: logs
17
+ mesh_format: obj
18
+ save_path: ???
19
+
20
+ ### Training
21
+ # guidance loss weights (0 to disable)
22
+ lambda_sd: 1
23
+ lambda_zero123: 0
24
+ # training batch size per iter
25
+ batch_size: 1
26
+ # training iterations for stage 1
27
+ iters: 500
28
+ # training iterations for stage 2
29
+ iters_refine: 50
30
+ # training camera radius
31
+ radius: 2.5
32
+ # training camera fovy
33
+ fovy: 49.1
34
+ # checkpoint to load for stage 1 (should be a ply file)
35
+ load:
36
+ # whether allow geom training in stage 2
37
+ train_geo: False
38
+ # prob to invert background color during training (0 = always black, 1 = always white)
39
+ invert_bg_prob: 0.5
40
+
41
+ ### GUI
42
+ gui: False
43
+ force_cuda_rast: False
44
+ # GUI resolution
45
+ H: 800
46
+ W: 800
47
+
48
+ ### Gaussian splatting
49
+ num_pts: 1000
50
+ sh_degree: 0
51
+ position_lr_init: 0.001
52
+ position_lr_final: 0.00002
53
+ position_lr_delay_mult: 0.02
54
+ position_lr_max_steps: 500
55
+ feature_lr: 0.01
56
+ opacity_lr: 0.05
57
+ scaling_lr: 0.005
58
+ rotation_lr: 0.005
59
+ percent_dense: 0.1
60
+ density_start_iter: 100
61
+ density_end_iter: 3000
62
+ densification_interval: 50
63
+ opacity_reset_interval: 700
64
+ densify_grad_threshold: 0.01
65
+
66
+ ### Textured Mesh
67
+ geom_lr: 0.0001
68
+ texture_lr: 0.2