2022-10-17 00:06:39 +08:00
|
|
|
# IGEV-Stereo & IGEV-MVS
|
2023-03-12 20:28:25 +08:00
|
|
|
This repository contains the source code for our paper:
|
2022-08-06 15:50:37 +08:00
|
|
|
|
2023-03-12 20:28:25 +08:00
|
|
|
Iterative Geometry Encoding Volume for Stereo Matching<br/>
|
|
|
|
CVPR 2023 <br/>
|
|
|
|
Gangwei Xu, Xianqi Wang, Xiaohuan Ding, Xin Yang<br/>
|
2022-08-06 15:50:37 +08:00
|
|
|
|
2023-03-12 20:37:24 +08:00
|
|
|
<img src="IGEV-Stereo/IGEV-Stereo.png">
|
|
|
|
|
2023-03-12 20:28:25 +08:00
|
|
|
|
|
|
|
#How to Use
|
2022-08-06 15:50:37 +08:00
|
|
|
|
|
|
|
## Environment
|
2023-03-12 20:28:25 +08:00
|
|
|
* NVIDIA RTX 3090
|
2022-08-06 15:50:37 +08:00
|
|
|
* Python 3.8
|
|
|
|
* Pytorch 1.12
|
|
|
|
|
|
|
|
## Install
|
|
|
|
|
|
|
|
### Create a virtual environment and activate it.
|
|
|
|
|
|
|
|
```
|
2022-09-27 14:45:53 +08:00
|
|
|
conda create -n IGEV_Stereo python=3.8
|
|
|
|
conda activate IGEV_Stereo
|
2022-08-06 15:50:37 +08:00
|
|
|
```
|
|
|
|
### Dependencies
|
|
|
|
|
|
|
|
```
|
|
|
|
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c nvidia
|
|
|
|
pip install opencv-python
|
|
|
|
pip install scikit-image
|
|
|
|
pip install tensorboard
|
|
|
|
pip install matplotlib
|
|
|
|
pip install tqdm
|
|
|
|
pip install timm==0.5.4
|
|
|
|
```
|