Awesome Open Source
Awesome Open Source

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch

This repository releases code for our paper DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch.

Table of Contents

Differentiable Patch Match
Requirements (Major Dependencies)


  • An efficient "Real Time Stereo Matching" algorithm, which takes as input 2 images and outputs a disparity (or depth) map.

  • Results/ Metrics:

    • KITTI: Results competitive to SOTA, while being real-time (8x faster than SOTA). SOTA among published real-time algorithms.

    • ETH3D: SOTA among all ROB entries.

    • SceneFlow: 2nd among all published algorithms, while being 8x faster than the 1st.

    • Runtime: 62ms (for DeepPruner-fast), 180ms (for DeepPruner-best)

    • Cuda Memory Requirements: 805MB (for DeepPruner-best)

Differentiable Patch Match

  • Fast algorithm for finding dense nearest neighbor correspondences between patches of images regions. Differentiable version of the generalized Patch Match algorithm. (Barnes et al.)

More details in the corresponding folder README.

Requirements (Major Dependencies)

  • Pytorch (0.4.1+)
  • Python2.7
  • torchvision (0.2.0+)


If you use our source code, or our paper, please consider citing the following:

title = {DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch},
author = {Shivam Duggal and Shenlong Wang and Wei-Chiu Ma and Rui Hu and Raquel Urtasun},
booktitle = {ICCV},
year = {2019} }

Correspondences to Shivam Duggal [email protected], Shenlong Wang [email protected], Wei-Chiu Ma [email protected]

Alternative Project Comparisons
Related Awesome Lists
Top Programming Languages

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Python (863,999
Learning (75,790
Algorithms (37,823
Pytorch (22,195
Patch (17,099
Real Time (10,657
Stereo Vision (276
Stereo Matching (91
Iccv2019 (86
Patchmatch (14