Update app.py
Browse files
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,
|
| 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()
|