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: