Skip to content
/ CSPN Public
forked from XinJCheng/CSPN

Convolutional Spatial Propagation Network

Notifications You must be signed in to change notification settings

yooooo00/CSPN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

78 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Convolutional Spatial Propagation Network

By Xinjing Cheng, Peng Wang, Chenye Guan and Ruigang Yang

Contents

  1. Introduction
  2. Citation

Introduction

This repo contains the CSPN models trained for depth completion and stereo depth estimation, as as described in the paper "Depth Estimation via Affinity Learned with Convolutional Spatial Propagation Network" and " Learning Depth with Convolutional Spatial Propagation Network"

We provide two implementations here. First, we implement a naive 2D CSPN by Pytorch which combine 'offsite' and 'conv' operations. Then, we provide a more efficient implementation of 2D/3D CSPN by CUDA and integrated them to the PaddlePaddle, which is more stable and easy to use.

Citation

If you use this method in your research, please cite:

@inproceedings{cheng2018depth,
  title={Depth estimation via affinity learned with convolutional spatial propagation network},
  author={Cheng, Xinjing and Wang, Peng and Yang, Ruigang},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  pages={103--119},
  year={2018}
}
@article{cheng2019learning,
  title={Learning depth with convolutional spatial propagation network},
  author={Cheng, Xinjing and Wang, Peng and Yang, Ruigang},
  journal={IEEE transactions on pattern analysis and machine intelligence},
  year={2019},
  publisher={IEEE}
}

About

Convolutional Spatial Propagation Network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.0%
  • Shell 1.0%