From 266b362e466aa0a48038bfe26e3d45c7fd378b45 Mon Sep 17 00:00:00 2001 From: Gangwei Xu Date: Mon, 20 Mar 2023 20:06:51 +0800 Subject: [PATCH] Update README.md --- README.md | 34 +++++++++++++--------------------- 1 file changed, 13 insertions(+), 21 deletions(-) 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