remove open3d

app.py

@@ -58,8 +58,6 @@ diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterizati
 os.system('pip install  dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html')
 # os.system('conda install conda-forge::ffmpeg')
 
-from dgl.geometry import farthest_point_sampler
-
 import sys
 sys.path.insert(0, str(Path(__file__).parent / "src"))
 sys.path.append(str(Path(__file__).parent / "src" / "experiments"))
@@ -88,6 +86,7 @@ def mat2quat(mat):
 
 
 def fps(x, enabled, n, device, random_start=False):
+    from dgl.geometry import farthest_point_sampler
     assert torch.diff(enabled * 1.0).sum() in [0.0, -1.0]
     start_idx = random.randint(0, enabled.sum() - 1) if random_start else 0
     fps_idx = farthest_point_sampler(x[enabled][None], n, start_idx=start_idx)[0]

src/experiments/real_world/gs/helpers.py

@@ -1,6 +1,6 @@
 import torch
 import os
-import open3d as o3d
+# import open3d as o3d
 import numpy as np
 from pathlib import Path
 import importlib
@@ -98,6 +98,7 @@ def quat_mult(q1, q2):
 
 
 def o3d_knn(pts, num_knn):
+    import open3d as o3d
     indices = []
     sq_dists = []
     pcd = o3d.geometry.PointCloud()
@@ -118,6 +119,7 @@ def o3d_knn(pts, num_knn):
     return np.array(sq_dists), np.array(indices)
 
 def o3d_knn_tensor(pts_tensor, num_knn):
+    import open3d as o3d
     if pts_tensor.numel() == 0:
         print("Point cloud is empty!")
         return None, None