HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Gangwei Xu 2023-03-20 20:06:51 +08:00 committed by GitHub
parent 25bd7f4fb0
commit 266b362e46
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
README.md
View File

@ -75,27 +75,6 @@ By default `stereo_datasets.py` will search for the datasets in these locations.
├── dtu_test ├── 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 ## Evaluation
To evaluate on Scene Flow or Middlebury or ETH3D, run 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 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 ## Submission
For submission to the KITTI benchmark, run For submission to the KITTI benchmark, run