/DySPN

Implement of "Dynamic Spatial Propagation Network for Depth Completion"

Primary LanguagePython

DySPN

Implement of "Dynamic Spatial Propagation Network for Depth Completion"

Paper: IEEE-TCSVT Early Version of AAAI (arxiv)

We will improve our repo as soon as possible.

Dependence

torch~=2.0.1

lightning~=2.0.1

Training

python3 lit_kitti_s.py

Acknowledgement

Thanks the authors for their awesome works: NLSPN CFormer

Citation

If you find our work useful in your research, please consider citing our paper:

@ARTICLE{lin2023dyspn,
  author={Lin, Yuankai and Yang, Hua and Cheng, Tao and Zhou, Wending and Yin, Zhouping},
  journal={IEEE Transactions on Circuits and Systems for Video Technology}, 
  title={DySPN: Learning Dynamic Affinity for Image-guided Depth Completion}, 
  year={2023},
  volume={},
  number={},
  pages={1-1},
  doi={10.1109/TCSVT.2023.3321668}}