hujiecpp commited on
Commit
8173ee2
·
1 Parent(s): bc287d4

init project

Browse files
Files changed (2) hide show
  1. app.py +4 -7
  2. 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)