Update README.md
This commit is contained in:
parent
6aa425678b
commit
7d0e60de3a
66
README.md
66
README.md
@ -33,8 +33,72 @@ pip install timm==0.5.4
|
||||
## Demos
|
||||
Pretrained models can be downloaded from [google drive](https://drive.google.com/drive/folders/1SsMHRyN7808jDViMN1sKz1Nx-71JxUuz?usp=share_link)
|
||||
|
||||
You can demo a trained model on Middlebury training pairs
|
||||
You can demo a trained model on pairs of images. To predict stereo for Middlebury, run
|
||||
```
|
||||
python demo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth
|
||||
```
|
||||
|
||||
|
||||
## Required Data
|
||||
To evaluate/train IGEV-Stereo, you will need to download the required datasets.
|
||||
* [Scene Flow Datasets](https://lmb.informatik.uni-freiburg.de/resources/datasets/SceneFlowDatasets.en.html)
|
||||
* [KITTI](http://www.cvlibs.net/datasets/kitti/eval_scene_flow.php?benchmark=stereo)
|
||||
* [Middlebury](https://vision.middlebury.edu/stereo/data/)
|
||||
* [ETH3D](https://www.eth3d.net/datasets#low-res-two-view-test-data)
|
||||
|
||||
By default `stereo_datasets.py` will search for the datasets in these locations.
|
||||
|
||||
```
|
||||
├── /data
|
||||
├── sceneflow
|
||||
├── frames_finalpass
|
||||
├── disparity
|
||||
├── KITTI
|
||||
├── KITTI_2012
|
||||
├── training
|
||||
├── testing
|
||||
├── vkitti
|
||||
├── KITTI_2015
|
||||
├── training
|
||||
├── testing
|
||||
├── vkitti
|
||||
├── Middlebury
|
||||
├── trainingH
|
||||
├── trainingH_GT
|
||||
├── ETH3D
|
||||
├── two_view_training
|
||||
├── two_view_training_gt
|
||||
```
|
||||
|
||||
|
||||
## 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
|
||||
```
|
||||
|
||||
## Submission
|
||||
|
||||
For submission to the KITTI benchmark, run
|
||||
```Shell
|
||||
python save_disp.py
|
||||
```
|
||||
|
||||
# Acknowledgements
|
||||
|
||||
This project is heavily based on [RAFT-Stereo](https://github.com/princeton-vl/RAFT-Stereo), We thank the original authors for their excellent work.
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user