Update mast3r/demo.py
Browse files- mast3r/demo.py +1 -1
mast3r/demo.py
CHANGED
@@ -15,6 +15,7 @@ import copy
|
|
15 |
from scipy.spatial.transform import Rotation
|
16 |
import tempfile
|
17 |
import shutil
|
|
|
18 |
|
19 |
from mast3r.cloud_opt.sparse_ga import sparse_global_alignment
|
20 |
from mast3r.cloud_opt.tsdf_optimizer import TSDFPostProcess
|
@@ -214,7 +215,6 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
|
|
214 |
|
215 |
def recon_fun(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
|
216 |
silent, image_size):
|
217 |
-
+import spaces
|
218 |
functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
|
219 |
silent, image_size)
|
220 |
+@spaces.GPU
|
|
|
15 |
from scipy.spatial.transform import Rotation
|
16 |
import tempfile
|
17 |
import shutil
|
18 |
+
+import spaces
|
19 |
|
20 |
from mast3r.cloud_opt.sparse_ga import sparse_global_alignment
|
21 |
from mast3r.cloud_opt.tsdf_optimizer import TSDFPostProcess
|
|
|
215 |
|
216 |
def recon_fun(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
|
217 |
silent, image_size):
|
|
|
218 |
functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
|
219 |
silent, image_size)
|
220 |
+@spaces.GPU
|