search

yzcjtr / GeoNet

Code for GeoNet: Unsupervised Learning of Dense Depth, Optical Flow and Camera Pose (CVPR 2018)
MIT License
726 stars 181 forks source link

PyTorch implementation? #22

Closed marianlonga closed 6 years ago

marianlonga commented 6 years ago

Hi, is there a PyTorch implementation of the paper instead of TensorFlow? Thanks!

yzcjtr commented 6 years ago

Hi, currently we don't have an official pytorch implementation yet. Though we welcome and appreciate any possible contributions. For reference, I would recommend these pytorch repos doing related tasks: SfmLearner, FlowNet2.