update

app.py

@@ -22,6 +22,11 @@ import kornia
 import yaml
 import matplotlib.pyplot as plt
 from sklearn.neighbors import NearestNeighbors
+
+gs_path = str(Path(__file__).parent / 'src/third-party/diff-gaussian-rasterization-w-depth')
+os.system(f'pip install -e {gs_path}')
+os.system('conda install -c dglteam/label/th24_cu124 dgl')
+os.system('conda install conda-forge::ffmpeg')
 from dgl.geometry import farthest_point_sampler
 
 import sys
@@ -60,6 +65,20 @@ def fps(x, enabled, n, device, random_start=False):
     return fps_idx
 
 
+def fps_new(x, enabled, n, device, random_start=False):
+    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
+    N = enabled.sum().item()
+    pcd = o3d.geometry.PointCloud()
+    pcd.points = o3d.utility.Vector3dVector(x[enabled].cpu().numpy())
+    pcd.colors = o3d.utility.Vector3dVector(
+        np.arange(len(pcd.points))[:, None] / len(pcd.points) * np.ones((1, 3), dtype=np.float32)
+    )
+    downpcd_farthest = pcd.farthest_point_down_sample(n)
+    fps_idx = np.array(downpcd_farthest.colors)[:, 0] * N
+    return fps_idx
+
+
 class DynamicsVisualizer:
 
     def __init__(self):

requirements.txt

@@ -27,4 +27,3 @@ torch==2.5.1 --index-url https://download.pytorch.org/whl/cu124
 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu124
 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
 torchdata==0.7.1
-dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html

src/experiments/log/gs/temp/gs_pred.splat

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:68163f3a942c74e9b71c679a84d8a1cd5be2bc9e054e19f5d350071a3746f500
+oid sha256:14a7bd4f83ff4c2c640e57589673f1c0b0ec571515f3bdb5437fba6bea9035ed
 size 7675392