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-12 20:49:36 +08:00 committed by GitHub
parent f1f4b791d4
commit 6aa425678b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
README.md
View File

@ -7,16 +7,11 @@ Gangwei Xu, Xianqi Wang, Xiaohuan Ding, Xin Yang<br/>
<img src="IGEV-Stereo/IGEV-Stereo.png"> <img src="IGEV-Stereo/IGEV-Stereo.png">
#How to Use
## Environment ## Environment
* NVIDIA RTX 3090 * NVIDIA RTX 3090
* Python 3.8 * Python 3.8
* Pytorch 1.12 * Pytorch 1.12
## Install
### Create a virtual environment and activate it. ### Create a virtual environment and activate it.
``` ```
@ -34,3 +29,12 @@ pip install matplotlib
pip install tqdm pip install tqdm
pip install timm==0.5.4 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
```
python demo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth
```