gaussian-splatting/docker/run.sh
2023-11-02 21:32:58 +02:00

31 lines
669 B
Bash
Executable File

#!/bin/bash
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$DIR"
#cd ..
mkdir -p $1-data/input
mkdir -p $1-data/output
FPS="5" #<-change this to change framerate
if [ -f $1-data/input/0001.jpg ]
then
echo "File $1 appears to have already been split .."
else
ffmpeg -i $1 -qscale:v 1 -qmin 1 -vf fps=$FPS $1-data/input/%04d.jpg
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/ --position_lr_init 0.000016 --scaling_lr 0.001 --iterations 100000 #Test more training budget
#pack it in
#tar cvfjh "$1.tar.bz2" $1-data/
exit 0