render_trex.sh

#!/bin/sh
#SBATCH --error=output/render_trex.error.%j  # STDOUT output is written in slurm.out.JOBID
#SBATCH --output=output/render_trex.out.%j # STDOUT error is written in slurm.err.JOBID
#SBATCH --job-name=render_trex_nsvf       # Job name
#SBATCH --mem=32GB                  # Memory request for this job
#SBATCH --nodes=1                   # The number of nodes
#SBATCH --partition=gpu-cluster
#SBATCH --account=vision
#SBATCH --time=48:0:0                # Runing time 2 days
#SBATCH --gpus=4                    # A number of GPUs
module load Anaconda3
module load CUDA/10.2
module load cuDNN/7
module load OpenMPI/3.1.4-GCC-8.3.0

. /ist/apps/modules/software/Anaconda3/5.3.0/etc/profile.d/conda.sh
conda activate venv_nsvf
DATA="trex_nsvf"
RES="756x1008"


ARCH="nsvf_base"
SUFFIX="v1"
DATASET=./data/${DATA}
SAVE=./runs/$DATA
MODEL=$ARCH$SUFFIX
MODEL_PATH=$SAVE/$MODEL/checkpoint_last.pt

MODELTEMP='{"chunk_size":%d,"raymarching_tolerance":%.3f,"use_octree":True}'
MODELARGS=$(printf "$MODELTEMP" 256 0.0)

python render.py ${DATASET} \
    --user-dir fairnr \
    --task single_object_rendering \
    --path ${MODEL_PATH} \
    --render-beam 1 \
    --render-save-fps 24 \
    --render-camera-poses $DATASET/test_traj.txt \
    --model-overrides $MODELARGS \
    --render-resolution $RES \
    --render-output ${SAVE}/output \
    --render-output-types "color" "depth" "voxel" "normal" \
    --render-combine-output --log-format "simple"