diff --git a/README.md b/README.md index b12e027..ff93b34 100644 --- a/README.md +++ b/README.md @@ -75,27 +75,6 @@ By default `stereo_datasets.py` will search for the datasets in these locations. ├── dtu_test ``` - -## Evaluation - -To evaluate a trained model on a test set (e.g. Scene Flow), run -```Shell -python evaluate_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset sceneflow -``` - -## Training - -To train on Scene Flow, run - -```Shell -python train_stereo.py -``` - -To train on KITTI, run -```Shell -python train_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset kitti -``` - ## Evaluation To evaluate on Scene Flow or Middlebury or ETH3D, run @@ -112,6 +91,19 @@ or python evaluate_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset eth3d ``` +## Training + +To train on Scene Flow, run + +```Shell +python train_stereo.py +``` + +To train on KITTI, run +```Shell +python train_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset kitti +``` + ## Submission For submission to the KITTI benchmark, run