Spaces:
Running
on
Zero
Running
on
Zero
test
Browse files
app.py
CHANGED
|
@@ -155,7 +155,7 @@ def run_dust3r(inputfiles, input_path=None):
|
|
| 155 |
return outfile, feat_image_path, opt, None, None
|
| 156 |
|
| 157 |
@spaces.GPU(duration=150)
|
| 158 |
-
def run_feat2gs(opt, niter=
|
| 159 |
|
| 160 |
if opt is None:
|
| 161 |
raise gr.Error("Please run Step 1 first!")
|
|
@@ -253,7 +253,7 @@ def run_render(opt, args, cam_traj='ellipse'):
|
|
| 253 |
def process_example(inputfiles, input_path):
|
| 254 |
dust3r_model, feat_image, dust3r_state, _, _ = run_dust3r(inputfiles, input_path=input_path)
|
| 255 |
|
| 256 |
-
output_model, feat2gs_state, _ = run_feat2gs(dust3r_state, niter=
|
| 257 |
|
| 258 |
output_video = run_render(dust3r_state, feat2gs_state, cam_traj='interpolated')
|
| 259 |
|
|
@@ -355,7 +355,7 @@ with demo:
|
|
| 355 |
|
| 356 |
with gr.Row(variant='panel'):
|
| 357 |
with gr.Tab("Step 2: Readout 3DGS from features & Jointly optimize pose"):
|
| 358 |
-
niter = gr.Number(value=
|
| 359 |
feat2gs_run = gr.Button("RUN Step 2")
|
| 360 |
with gr.Column(scale=1):
|
| 361 |
with gr.Group():
|
|
|
|
| 155 |
return outfile, feat_image_path, opt, None, None
|
| 156 |
|
| 157 |
@spaces.GPU(duration=150)
|
| 158 |
+
def run_feat2gs(opt, niter=200):
|
| 159 |
|
| 160 |
if opt is None:
|
| 161 |
raise gr.Error("Please run Step 1 first!")
|
|
|
|
| 253 |
def process_example(inputfiles, input_path):
|
| 254 |
dust3r_model, feat_image, dust3r_state, _, _ = run_dust3r(inputfiles, input_path=input_path)
|
| 255 |
|
| 256 |
+
output_model, feat2gs_state, _ = run_feat2gs(dust3r_state, niter=200)
|
| 257 |
|
| 258 |
output_video = run_render(dust3r_state, feat2gs_state, cam_traj='interpolated')
|
| 259 |
|
|
|
|
| 355 |
|
| 356 |
with gr.Row(variant='panel'):
|
| 357 |
with gr.Tab("Step 2: Readout 3DGS from features & Jointly optimize pose"):
|
| 358 |
+
niter = gr.Number(value=200, precision=0, minimum=1000, maximum=8000, label="Training iterations")
|
| 359 |
feat2gs_run = gr.Button("RUN Step 2")
|
| 360 |
with gr.Column(scale=1):
|
| 361 |
with gr.Group():
|