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():
|