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