uber-research/DeepPruner uber-research/DeepPruner

  • Stars
    star
    351
  • Rank 120,906 (Top 3 %)
  • Language
    Python
  • License
    Other
  • Created about 5 years ago
  • Updated about 4 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
uber-research/DeepPruner
github

uber-research

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

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

DeepPruner
Differentiable Patch Match
Requirements (Major Dependencies)
Citation

DeepPruner

  • 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+)

Citation

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

@inproceedings{Duggal2019ICCV,
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]

More Repositories

1

deep-neuroevolution

Deep Neuroevolution
Python
1,630
star
2

PPLM

Plug and Play Language Model implementation. Allows to steer topic and attributes of GPT-2 models.
Python
1,125
star
3

UPSNet

UPSNet: A Unified Panoptic Segmentation Network
Python
639
star
4

go-explore

Code for Go-Explore: a New Approach for Hard-Exploration Problems
Python
553
star
5

PyTorch-NEAT

Python
526
star
6

LaneGCN

[ECCV2020 Oral] Learning Lane Graph Representations for Motion Forecasting
Python
502
star
7

sbnet

Sparse Blocks Networks
Python
430
star
8

differentiable-plasticity

Implementations of the algorithms described in Differentiable plasticity: training plastic networks with gradient descent, a research paper from Uber AI Labs.
Python
394
star
9

parallax

Tool for interactive embeddings visualization
Python
284
star
10

learning-to-reweight-examples

Code for paper "Learning to Reweight Examples for Robust Deep Learning"
Python
269
star
11

jpeg2dct

C++
251
star
12

poet

Paired Open-Ended Trailblazer (POET) and Enhanced POET
Python
235
star
13

intrinsic-dimension

Jupyter Notebook
220
star
14

CoordConv

Python
208
star
15

atari-model-zoo

A binary release of trained deep reinforcement learning models trained in the Atari machine learning benchmark, and a software release that enables easy visualization and analysis of models, and comparison across training algorithms.
Jupyter Notebook
201
star
16

ape-x

This repo replicates the results Horgan et al obtained in "Distributed Prioritized Experience Replay"
Python
188
star
17

EvoGrad

Python
178
star
18

TuRBO

Python
178
star
19

safemutations

safemutations
C++
143
star
20

permute-quantize-finetune

Using ideas from product quantization for state-of-the-art neural network compression.
Python
143
star
21

deconstructing-lottery-tickets

Python
142
star
22

CRISP

Python
131
star
23

metropolis-hastings-gans

Python
112
star
24

GTN

Python
75
star
25

backpropamine

Train self-modifying neural networks with neuromodulated plasticity
Python
73
star
26

loss-change-allocation

Python
61
star
27

MARVIN

Uber's Multi-Agent Routing Value Iteration Network
Python
57
star
28

GOCC

Go
51
star
29

Synthetic-Petri-Dish

Python
42
star
30

RxThreadEffectChecker

Static checker for Rx Threading Effects, based on the Checker Framework
Java
35
star
31

Map-Elites-Evolutionary

Map-Elites based on Evolution Strategies
Python
31
star
32

D3G

Estimating Q(s,s') with Deep Deterministic Dynamics Gradients
Python
29
star
33

java-dependency-validator

Dependency validator detects runtime compatibility issues at build time
Java
23
star
34

vargp

Variational Auto-Regressive Gaussian Processes for Continual Learning
Python
20
star
35

normative-uncertainty

Python
15
star
36

Evolvability-ES

Python
14
star
37

brezel

Starlark
8
star
38

dispatch-optim

Constrainted based optimization
Python
8
star
39

ga-world-models

Python
7
star
40

FSDM

Code tor the SIGDIAL 2019 paper Flexibly-Structured Model for Task-Oriented Dialogues. It implements a deep learning end-to-end differentiable dialogue system model
Python
7
star
41

rl-controller-verification

Quadcopter Verification
Python
6
star
42

go-context-propagate

Go
4
star
43

last-diff-analyzer

A multi-language tool for checking semantic equivalence for code
Go
2
star
44

presto-HDFS-read-data

A dump of some of our Presto logs, for use as part of ongoing Presto/HDFS research and presentations.
2
star
45

xplane-bazel-docker

Bazel Xplane
C++
1
star
46

tailr

TAILR
Python
1
star