Spaces:
Running
on
Zero
Running
on
Zero
Reduce gpu time and iterations for stage 1.
Browse files- 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=
|
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 =
|
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,
|
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 |
],
|