bartduis commited on
Commit
dffb3df
·
verified ·
1 Parent(s): 993f92c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ def rayst3r_to_glb(img,depth_dict,mask,max_total_points=10e6,rotated=False):
114
  prep_for_rayst3r(img,depth_dict,mask)
115
 
116
  rayst3r_points = eval_scene(rayst3r_model,os.path.join(outdir, "input"),do_filter_all_masks=True,dino_model=dino_model, device = device).cpu()
117
-
118
  # subsample points
119
  n_points = min(max_total_points,rayst3r_points.shape[0])
120
  rayst3r_points = rayst3r_points[torch.randperm(rayst3r_points.shape[0])[:n_points]].numpy()
 
114
  prep_for_rayst3r(img,depth_dict,mask)
115
 
116
  rayst3r_points = eval_scene(rayst3r_model,os.path.join(outdir, "input"),do_filter_all_masks=True,dino_model=dino_model, device = device).cpu()
117
+ print(rayst3r_points)
118
  # subsample points
119
  n_points = min(max_total_points,rayst3r_points.shape[0])
120
  rayst3r_points = rayst3r_points[torch.randperm(rayst3r_points.shape[0])[:n_points]].numpy()