hujiecpp commited on
Commit
c0c5360
·
1 Parent(s): eb0c94f

init project

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,10 +38,9 @@ from modules.pe3r.models import Models
38
  import torchvision.transforms as tvf
39
 
40
  silent = False
41
- device = 'cuda' if torch.cuda.is_available() else 'cpu'
42
  pe3r = Models(device) #
43
 
44
-
45
  def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
46
  cam_color=None, as_pointcloud=False,
47
  transparent_cams=False):
@@ -433,6 +432,7 @@ def get_cog_feats(images):
433
 
434
  return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
435
 
 
436
  @spaces.GPU(duration=120)
437
  def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
438
  as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
 
38
  import torchvision.transforms as tvf
39
 
40
  silent = False
41
+ device = 'cpu' # 'cuda' if torch.cuda.is_available() else
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):
 
432
 
433
  return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
434
 
435
+
436
  @spaces.GPU(duration=120)
437
  def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
438
  as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,