hujiecpp commited on
Commit
e53d99d
·
1 Parent(s): 7398aa3

init project

Browse files
Files changed (1) hide show
  1. modules/pe3r/demo.py +3 -4
modules/pe3r/demo.py CHANGED
@@ -424,7 +424,7 @@ def get_cog_feats(images, pe3r, device):
424
 
425
  return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
426
 
427
-
428
  def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, niter, min_conf_thr,
429
  as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
430
  scenegraph_type, winsize, refid):
@@ -511,7 +511,7 @@ def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, ni
511
 
512
  return scene, outfile, imgs
513
 
514
-
515
  def get_3D_object_from_scene(outdir, pe3r, silent, text, threshold, scene, min_conf_thr, as_pointcloud,
516
  mask_sky, clean_depth, transparent_cams, cam_size):
517
 
@@ -549,10 +549,9 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
549
  return winsize, refid
550
 
551
 
552
- @spaces.GPU(duration=180)
553
  def main_demo(tmpdirname, server_name, server_port, silent=False):
554
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
555
-
556
  pe3r = Models(device)
557
  # scene, outfile, imgs = get_reconstructed_scene(
558
  # outdir=tmpdirname, pe3r=pe3r, device=device, silent=silent,
 
424
 
425
  return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
426
 
427
+ @spaces.GPU(duration=180)
428
  def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, niter, min_conf_thr,
429
  as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
430
  scenegraph_type, winsize, refid):
 
511
 
512
  return scene, outfile, imgs
513
 
514
+ @spaces.GPU(duration=180)
515
  def get_3D_object_from_scene(outdir, pe3r, silent, text, threshold, scene, min_conf_thr, as_pointcloud,
516
  mask_sky, clean_depth, transparent_cams, cam_size):
517
 
 
549
  return winsize, refid
550
 
551
 
 
552
  def main_demo(tmpdirname, server_name, server_port, silent=False):
553
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
554
+
555
  pe3r = Models(device)
556
  # scene, outfile, imgs = get_reconstructed_scene(
557
  # outdir=tmpdirname, pe3r=pe3r, device=device, silent=silent,