Download Scene Flow Datasets, KITTI 2012, KITTI 2015
Please update DATAPATH in the sceneflow.sh file as your training data path.
Multi-GPU training
export NGPUS=8
export save_results_dir="./checkpoints"
./scripts/sceneflow.sh $NGPUS $save_results_dir --dlossw "1.0,0.5" --batch_size 2 --eval_freq 3 --model mynet
- Set
CKPT_FILEas your checkpoint file
export NGPUS=8
export save_results_dir="./checkpoints"
./scripts/sceneflow.sh $NGPUS $save_results_dir --loadckpt $CKPT_FILE- "4,1" --batch_size 2 --mode test --model mynet
Please update DATAPATH in the kitti15.sh file as your training data path.
export NGPUS=8
export save_results_dir="./checkpoints"
./scripts/kitti15.sh $NGPUS $save_results_dir --dlossw "0.5,1.0" --batch_size 2 --eval_freq 3 --model mynet
- Set
CKPT_FILEas your checkpoint file
export NGPUS=8
export save_results_dir="./checkpoints"
./scripts/kitti15.sh $NGPUS $save_results_dir --loadckpt $CKPT_FILE- --batch_size 2 --mode test --model mynet
export save_path="./outputs/predictions"
./scripts/kitti15_save.sh $save_results_dir --loadckpt $CKPT_FILE --batch_size 2 --model mynet