UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2025-02-17 01:37:04 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

....

Browse Source
This commit is contained in:
Ammar Qammaz 2023-10-25 18:57:21 +03:00
parent 74b996fcc3
commit e16749ab9b
3 changed files with 30 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile
View File

@ -13,6 +13,8 @@ RUN \
vim \
build-essential \
cmake \
ffmpeg \
freeglut3-dev \
libopencv-dev \
libjpeg-dev \
libpng-dev \
@ -20,6 +22,8 @@ RUN \
libpthread-stubs0-dev \
git \
virtualenv \
sqlite3 \
libsqlite3-dev \
time \
sudo \
wget \

12
docker/initialize.sh
View File

@ -1,12 +1,16 @@
#!/bin/bash
#git clone https://github.com/NVIDIA/cuda-samples
#cd cuda-samples
git clone https://ceres-solver.googlesource.com/ceres-solver
cd ceres-solver
git checkout $(git describe --tags) # Checkout the latest release
mkdir build
cd build
cmake .. -DBUILD_TESTING=OFF -DBUILD_EXAMPLES=OFF -DCMAKE_CUDA_ARCHITECTURES=native
cmake .. -DBUILD_TESTING=OFF -DBUILD_EXAMPLES=OFF -DUSE_CUDA=OFF #-DCMAKE_CUDA_ARCHITECTURES=native
make
sudo make install
cd ..
@ -18,7 +22,7 @@ cd colmap
git checkout dev
mkdir build
cd build
cmake .. -DCMAKE_CUDA_ARCHITECTURES=native
cmake .. -DCMAKE_CUDA_ARCHITECTURES=native -DCUDA_ENABLED=OFF
make
sudo make install
CC=/usr/bin/gcc-6 CXX=/usr/bin/g++-6 cmake ..
@ -28,7 +32,7 @@ cd ..
pip install -q plyfile
pip install -q https://huggingface.co/camenduru/gaussian-splatting/resolve/main/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
pip install -q https://huggingface.co/camenduru/gaussian-splatting/resolve/main/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl
python3.10 -m pip install -q https://huggingface.co/camenduru/gaussian-splatting/resolve/main/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
python3.10 -m pip install -q https://huggingface.co/camenduru/gaussian-splatting/resolve/main/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl
exit 0

18
docker/run.sh Executable file
View File

@ -0,0 +1,18 @@
#!/bin/bash
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$DIR"
cd ..
mkdir -p $1-data/input
mkdir -p $1-data/output
ffmpeg -i $1 -qscale:v 1 -qmin 1 -vf fps={fps} $1-data/input/%04d.jpg
python3.10 convert.py -s $1-data/ --no_gpu
python3.10 train.py -s $1-data/ -r 1 --model_path=$1-data/output/
exit 0