Spaces:
Runtime error
Runtime error
Upload 2 files
Browse files- configs/image.yaml +69 -0
- 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
|