Skip to content

Latest commit

 

History

History
 
 

gen2-crestereo-stereo-matching

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
models
models
 
 
README.md
README.md
 
 
download.py
download.py
 
 
main.py
main.py
 
 
requirements.txt
requirements.txt
 
 

README.md

CREStereo

CREStereo: Practical Stereo Matching via Cascaded Recurrent Network with Adaptive Correlation

Demo

160x240

python3 main.py -nn models/crestereo_init_iter2_160x240_6_shaves.blob -shape 160x240

demo-gif

120x160

python3 main.py -nn models/crestereo_init_iter2_120x160_6_shaves.blob -shape 120x160

demo-gif

Install requirements

python3 -m pip install -r requirements.txt

Download blobs:

python3 download.py

Usage

Note: Make sure -shape matches the blob's model input shape

usage: main.py [-h] -nn NN_PATH -shape {120x160,160x240}

optional arguments:
  -h, --help            show this help message and exit
  -nn NN_PATH, --nn_path NN_PATH
                        select model blob path for inference
  -shape {120x160,160x240}, --shape {120x160,160x240}
                        model input shape, same as used blob

References