Skip to content

CUDA implementation of Stereo Marching for Pytorch(Support forward and backward). building...

Notifications You must be signed in to change notification settings

lzhnb/CuStereoMatching

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CuStereoMatching

CuStereoMatching is an CUDA implementation of stereo matching.

Requirements

The enviroment of my developer machine:

  • Python 3.8.8+
  • PyTorch 1.10.2
  • CUDA 11.1

Installation

python setup.py install

Or use:

pip install .

Or use:

pip install https://github.com/lzhnb/CuStereoMatching

TODO

  • Examples (More Example)
  • Support backward
  • Optimize the code
  • More elegant Python Wrapper
  • Visualization

Example

Put the points.npy file under examples directory, then run

python examples/verify.py

About

CUDA implementation of Stereo Marching for Pytorch(Support forward and backward). building...

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published