faneggg commited on
Commit
e502ae4
·
1 Parent(s): dc7576a
Files changed (1) hide show
  1. app.py +3 -3
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=2000):
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=2000)
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=2000, 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():
 
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():