diff --git a/docker/initialize.sh b/docker/initialize.sh index 007ef11..b9753fa 100755 --- a/docker/initialize.sh +++ b/docker/initialize.sh @@ -8,6 +8,7 @@ #also get add-ons git clone https://github.com/antimatter15/splat git clone https://github.com/ReshotAI/gaussian-splatting-blender-addon/ +git clone https://github.com/francescofugazzi/3dgsconverter #needs scikit-learn @@ -37,7 +38,7 @@ cd .. -python3.10 -m pip install plyfile tqdm +python3.10 -m pip install plyfile tqdm scikit-learn python3.10 -m pip install 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 https://huggingface.co/camenduru/gaussian-splatting/resolve/main/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl python3.10 -m pip install torchvision diff --git a/docker/run.sh b/docker/run.sh index 81f6d76..834e1f0 100755 --- a/docker/run.sh +++ b/docker/run.sh @@ -23,6 +23,9 @@ fi python3.10 convert.py -s $1-data/ --camera SIMPLE_RADIAL --no_gpu #GPU produces worse results (?) python3.10 train.py -s $1-data/ -r 1 --model_path=$1-data/output/ # +python3.10 3dgsconverter/3dgsconverter.py -i $1-data/output/point_cloud/iteration_30000/point_cloud.ply -o $1-data/output/point_cloud/iteration_30000/output_cc.ply -f cc --rgb --density_filter --remove_flyers +python3.10 3dgsconverter/3dgsconverter.py -i $1-data/output/output/point_cloud/iteration_30000/output_cc.ply -o $1-data/output/output/point_cloud/iteration_30000/point_cloud_clean.ply -f 3dgs + #pack it in #tar cvfjh "$1.tar.bz2" $1-data/