init project
Browse files- app.py +4 -7
- modules/pe3r/demo.py +0 -1
app.py
CHANGED
@@ -37,6 +37,10 @@ import matplotlib.pyplot as pl
|
|
37 |
from modules.mobilesamv2.utils.transforms import ResizeLongestSide
|
38 |
from modules.pe3r.models import Models
|
39 |
|
|
|
|
|
|
|
|
|
40 |
def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
|
41 |
cam_color=None, as_pointcloud=False,
|
42 |
transparent_cams=False, silent=False):
|
@@ -552,13 +556,6 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
|
|
552 |
return winsize, refid
|
553 |
|
554 |
|
555 |
-
|
556 |
-
|
557 |
-
|
558 |
-
silent = True
|
559 |
-
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
560 |
-
pe3r = Models(device)
|
561 |
-
|
562 |
with tempfile.TemporaryDirectory(suffix='pe3r_gradio_demo') as tmpdirname:
|
563 |
recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, pe3r, device, silent)
|
564 |
model_from_scene_fun = functools.partial(get_3D_model_from_scene, tmpdirname, silent)
|
|
|
37 |
from modules.mobilesamv2.utils.transforms import ResizeLongestSide
|
38 |
from modules.pe3r.models import Models
|
39 |
|
40 |
+
silent = True
|
41 |
+
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
42 |
+
pe3r = Models(device)
|
43 |
+
|
44 |
def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
|
45 |
cam_color=None, as_pointcloud=False,
|
46 |
transparent_cams=False, silent=False):
|
|
|
556 |
return winsize, refid
|
557 |
|
558 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
559 |
with tempfile.TemporaryDirectory(suffix='pe3r_gradio_demo') as tmpdirname:
|
560 |
recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, pe3r, device, silent)
|
561 |
model_from_scene_fun = functools.partial(get_3D_model_from_scene, tmpdirname, silent)
|
modules/pe3r/demo.py
CHANGED
@@ -550,7 +550,6 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
|
|
550 |
|
551 |
def main_demo(tmpdirname, server_name, server_port, silent=False):
|
552 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
553 |
-
|
554 |
pe3r = Models(device)
|
555 |
|
556 |
recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, pe3r, device, silent)
|
|
|
550 |
|
551 |
def main_demo(tmpdirname, server_name, server_port, silent=False):
|
552 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
553 |
pe3r = Models(device)
|
554 |
|
555 |
recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, pe3r, device, silent)
|