CSPN implemented in Pytorch 0.4.1

Introduction

This is a PyTorch(0.4.1) implementation of Depth Estimation via Affinity Learned with Convolutional Spatial Propagation Network. At present, we can provide train script in NYU Depth V2 dataset for depth completion and monocular depth estimation. KITTI will be available soon!

The implementation of multi-gpus is based on Pytorch Encoding.