Awesome

EAI-Stereo

Software Requirements

PyTorch 1.12.0
CUDA 11.7

pip install scipy
pip install tqdm
pip install tensorboard
pip install opt_einsum
pip install imageio
pip install opencv-python
pip install scikit-image

Required Data

To evaluate/train EAI-Stereo, you will need to download the required datasets.

• Sceneflow
• Middlebury
• ETH3D
• KITTI

By default stereo_datasets.py will search for the datasets in these locations. You can create symbolic links to wherever the datasets were downloaded in the datasets folder

├── datasets
    ├── FlyingThings3D
        ├── frames_cleanpass
        ├── frames_finalpass
        ├── disparity
    ├── Monkaa
        ├── frames_cleanpass
        ├── frames_finalpass
        ├── disparity
    ├── Driving
        ├── frames_cleanpass
        ├── frames_finalpass
        ├── disparity
    ├── KITTI
        ├── testing
        ├── training
        ├── devkit
    ├── Middlebury
        ├── MiddEval3
    ├── ETH3D
        ├── two_view_testing

Build sampler

cd sampler && python setup.py install && cd ..

Train

bash ./train.sh

Evaluate

Set the arguments in evaluate_stereo.py and execute

python evaluate_stereo.py