Add design file

app.py

@@ -187,6 +187,7 @@ def local_get_reconstructed_scene(inputfiles, min_conf_thr, cam_size):
     ])
     os.system('pip install iopath')
     os.system(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html")
+    from pytorch3d.ops import knn_points
     outdir = tempfile.mkdtemp(suffix='_FLARE_gradio_demo')
     batch = load_images(inputfiles, size=image_size, verbose=not silent)
     images = [gt['img'] for gt in batch]
@@ -270,11 +271,11 @@ def local_get_reconstructed_scene(inputfiles, min_conf_thr, cam_size):
     # use knn to clean the point cloud
     K = 10
     points = torch.tensor(pts3d.reshape(1,-1,3)).cuda()
-    # knn = knn_points(points, points, K=K)
-    # dists = knn.dists  
-    # mean_dists = dists.mean(dim=-1)
-    # masks_dist = mean_dists < torch.quantile(mean_dists.reshape(-1), 0.95)
-    # masks_dist = masks_dist.detach().cpu().numpy()
+    knn = knn_points(points, points, K=K)
+    dists = knn.dists  
+    mean_dists = dists.mean(dim=-1)
+    masks_dist = mean_dists < torch.quantile(mean_dists.reshape(-1), 0.95)
+    masks_dist = masks_dist.detach().cpu().numpy()
     # masks_conf = (masks_conf > 0# & masks_dist.reshape(-1,H,W)
     masks_conf = masks_conf > 0
     os.makedirs(outdir, exist_ok=True)