ostapagon commited on
Commit
f8adcbc
·
1 Parent(s): 1fe4c34

Reduce gpu time and iterations for stage 1.

Browse files
Files changed (1) hide show
  1. demo/mast3r_demo.py +3 -3
demo/mast3r_demo.py CHANGED
@@ -173,7 +173,7 @@ def save_colmap_scene(scene, save_dir, min_conf_thr=2, clean_depth=False, mask_i
173
  save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
174
  return save_path
175
 
176
- @spaces.GPU(duration=170)
177
  def get_reconstructed_scene(snapshot, current_scene_state,
178
  min_conf_thr, matching_conf_thr,
179
  as_pointcloud, cam_size, shared_intrinsics, clean_depth, filelist, **kw):
@@ -189,7 +189,7 @@ def get_reconstructed_scene(snapshot, current_scene_state,
189
  filelist = [filelist[0], filelist[0] + '_2']
190
 
191
  lr1 = 0.07
192
- niter1 = 700
193
  lr2 = 0.014
194
  niter2 = 300
195
  optim_level = 'refine+depth'
@@ -346,7 +346,7 @@ def mast3r_demo_tab():
346
  [
347
  tower_images[0],
348
  None,
349
- 1.5, 0.0, 0.2, True, True, False,
350
  tower_images,
351
  ]
352
  ],
 
173
  save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
174
  return save_path
175
 
176
+ @spaces.GPU(duration=160)
177
  def get_reconstructed_scene(snapshot, current_scene_state,
178
  min_conf_thr, matching_conf_thr,
179
  as_pointcloud, cam_size, shared_intrinsics, clean_depth, filelist, **kw):
 
189
  filelist = [filelist[0], filelist[0] + '_2']
190
 
191
  lr1 = 0.07
192
+ niter1 = 600
193
  lr2 = 0.014
194
  niter2 = 300
195
  optim_level = 'refine+depth'
 
346
  [
347
  tower_images[0],
348
  None,
349
+ 1.5, 0.0, 0.2, True, False, False,
350
  tower_images,
351
  ]
352
  ],