chore: Update pixi.toml with additional channels and dependencies

pixi.toml

@@ -3,7 +3,7 @@ name = "mini-dpvo"
 version = "0.1.1"
 description = "Add a short description here"
 authors = ["pablovela5620 <[email protected]>"]
-channels = ["nvidia/label/cuda-11.8.0", "nvidia", "conda-forge", "pytorch"]
+channels = ["nvidia/label/cuda-11.8.0", "nvidia", "conda-forge", "pytorch", "pyg"]
 platforms = ["linux-64"]
 
 [system-requirements]
@@ -19,11 +19,20 @@ cuda = {version = "*", channel="nvidia/label/cuda-11.8.0"}
 pytorch-cuda = {version = "11.8.*", channel="pytorch"}
 pytorch = {version = ">=2.2.0,<2.3", channel="pytorch"}
 torchvision = {version = ">=0.17.0,<0.18", channel="pytorch"}
-einops = ">=0.7.0,<0.8"
+plyfile = ">=1.0.3,<1.1"
+pytorch-scatter = {version="*", channel="pyg"}
+matplotlib = ">=3.8.4,<3.9"
+yacs = ">=0.1.8,<0.2"
 jaxtyping = ">=0.2.28,<0.3"
-gradio = ">=4.29.0,<4.30"
-rerun-sdk = ">=0.15.1,<0.16"
+icecream = ">=2.1.3,<2.2"
+rerun-sdk = "0.15.*"
+gradio = "4.31.2.*"
+eigen = ">=3.4.0,<3.5"
 
 [pypi-dependencies]
 mini-dust3r = "*"
-spaces = "==0.28.3"
+spaces = "==0.28.3"
+opencv-python = ">=4.9.0.80"
+evo = ">=1.28.0"
+gradio-rerun = "*"
+mmcv = "*"

setup.py

@@ -0,0 +1,49 @@
+import os.path as osp
+from setuptools import setup, find_packages
+from torch.utils.cpp_extension import BuildExtension, CUDAExtension
+
+ROOT = osp.dirname(osp.abspath(__file__))
+
+
+setup(
+    name="mini_dpvo",
+    packages=find_packages(),
+    ext_modules=[
+        CUDAExtension(
+            "cuda_corr",
+            sources=[
+                "mini_dpvo/altcorr/correlation.cpp",
+                "mini_dpvo/altcorr/correlation_kernel.cu",
+            ],
+            extra_compile_args={
+                "cxx": ["-O3"],
+                "nvcc": ["-O3"],
+            },
+        ),
+        CUDAExtension(
+            "cuda_ba",
+            sources=["mini_dpvo/fastba/ba.cpp", "mini_dpvo/fastba/ba_cuda.cu"],
+            extra_compile_args={
+                "cxx": ["-O3"],
+                "nvcc": ["-O3"],
+            },
+        ),
+        CUDAExtension(
+            "lietorch_backends",
+            include_dirs=[
+                osp.join(ROOT, "mini_dpvo/lietorch/include"),
+                osp.join(ROOT, ".pixi/envs/default/include/eigen3"),
+            ],
+            sources=[
+                "mini_dpvo/lietorch/src/lietorch.cpp",
+                "mini_dpvo/lietorch/src/lietorch_gpu.cu",
+                "mini_dpvo/lietorch/src/lietorch_cpu.cpp",
+            ],
+            extra_compile_args={
+                "cxx": ["-O3"],
+                "nvcc": ["-O3"],
+            },
+        ),
+    ],
+    cmdclass={"build_ext": BuildExtension},
+)