bartduis commited on
Commit
d2e8abe
·
verified ·
1 Parent(s): 38add22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def prep_for_rayst3r(img,depth_dict,mask):
113
  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,do_filter_all_masks=True).cpu()
117
  # subsample points
118
  n_points = min(max_total_points,rayst3r_points.shape[0])
119
  rayst3r_points = rayst3r_points[torch.randperm(rayst3r_points.shape[0])[:n_points]].numpy()
 
113
  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,set_conf=10).cpu()
117
  # subsample points
118
  n_points = min(max_total_points,rayst3r_points.shape[0])
119
  rayst3r_points = rayst3r_points[torch.randperm(rayst3r_points.shape[0])[:n_points]].numpy()