下載源碼:git clone --recurse-submodules https://github.com/graphdeco-inria/gaussian-splatting.git
利用conda創建虛擬環境:
conda create -n gs python=3.8
切換虛擬環境:
conda activate gs
在gs環境下安裝:
pip install torch==2.0.0+cu118 torchvision==0.15.0+cu118 torchaudio==2.0.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html
pip install plyfile
pip install tqdm進
入項目文件夾:
cd gaussian-splatting
pip install submodules/diff-gaussian-rasterization/
pip install submodules/simple-knn/
訓練:
python train.py -s data/tandt/train
python train.py -s data/db/playroom