From 150b4877c4be5cc587656d5036b8214b42816ad9 Mon Sep 17 00:00:00 2001 From: Gangwei Xu Date: Mon, 20 Mar 2023 20:02:42 +0800 Subject: [PATCH] Update README.md --- README.md | 34 +++++++++++++++++++++++++++++++++- 1 file changed, 33 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index ccd7411..b12e027 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,4 @@ # IGEV-Stereo & IGEV-MVS -The IGEV-MVS is coming soon. This repository contains the source code for our paper: @@ -71,6 +70,9 @@ By default `stereo_datasets.py` will search for the datasets in these locations. ├── ETH3D ├── two_view_training ├── two_view_training_gt + ├── DTU_data + ├── dtu_train + ├── dtu_test ``` @@ -94,6 +96,22 @@ To train on KITTI, run python train_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset kitti ``` +## Evaluation + +To evaluate on Scene Flow or Middlebury or ETH3D, run + +```Shell +python evaluate_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset sceneflow +``` +or +```Shell +python evaluate_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset middlebury_H +``` +or +```Shell +python evaluate_stereo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth --dataset eth3d +``` + ## Submission For submission to the KITTI benchmark, run @@ -101,6 +119,20 @@ For submission to the KITTI benchmark, run python save_disp.py ``` +## MVS training and evaluation + +To train on DTU, run + +```Shell +python train_mvs.py +``` + +To evaluate on DTU, run + +```Shell +python evaluate_mvs.py +``` + ## Citation If you find our work useful in your research, please consider citing our paper: