requirements.txt

-f https://download.pytorch.org/whl/torch_stable.html
-f https://data.pyg.org/whl/torch-2.0.1+cu118.html
albumentations==0.5.2
astral==2.2
astropy==5.1
attrdict==2.0.1
beartype==0.11.0
colorlog==6.6.0
coverage==6.2
cstruct==3.1
cycler==0.10.0
dask==2022.9.1
decorator==4.4.2
deepdiff==5.8.0
einops==0.4.1
geopandas==0.12.2
gitpython==2.0.0
hyperopt==0.2.7
ijson==3.1.4
imageio==2.9.0
imgaug==0.4.0
jinja2==3.0.3
joblib==1.1.1
jsonpath-ng==1.5.3
kiwisolver==1.3.1
kornia
laspy==2.0.3
laspy[laszip]
laszip==0.1.0
matplotlib==3.5.1
more-itertools==8.14.0
nptyping==2.3.1
numba==0.56.4
numpy==1.23.5
numpy-quaternion==2022.4.3
omegaconf==2.2.3
onnx==1.13.1
onnx-simplifier==0.3.6
onnxruntime==1.15.1 #nnef-tools==1.0 is not compatible with greater versions
open3d==0.16.0
opencv-python==4.7.0.72
opencv-python-headless==4.7.0.72
openpyxl==3.1.1
orjson==3.8.1
pandas==1.4.3
pathos==0.2.9
pickle5
Pillow==9.5.0
protobuf==3.20.3
psutil==5.9.0
pytorch-metric-learning
PyTurboJPEG
PyWavelets==1.4.1
PyYAML==6.0
scikit-learn==1.2.1
scipy==1.10.1
seaborn==0.11.2
Shapely==2.0.1
six==1.16.0
sortedcontainers==2.4.*
tensorboard==2.13.*
tensordict
torch==2.1.1+cu118
torch-scatter==2.1.1+pt20cu118
torchaudio
torchgeometry
torchmetrics
torchvision
tqdm==4.62.3
typeguard==2.13.3
av2
bpy==4.0.0
mathutils==3.3.0
pyquaternion==0.9.9
gsplat==1.0.0
plyfile