Implementation of LiteFlowNet with tensorflow
keeper121 opened this issue · 4 comments
Hi, @twhui. I try to implement LiteFlowNet's inference with tensorflow. I convert pretrained weights from https://github.com/sniklaus/pytorch-liteflownet (pytorch reimplementation). Seems it produce a bit different optical flow. I think it depends on difference in warping layer. But will be cool if you can add link to my repo.
Please take a look.
https://github.com/keeper121/liteflownet-tf2
Hi, @keeper121 . Thanks for your implementation of my work in tensorflow. I would suggest you to include a link (https://github.com/twhui/LiteFlowNet#license-and-citation) for the licensing terms of my work.
The current tensorflow implementation requires the conversion of trained model from Caffe to Pytorch and then to Tensorflow. Could you simply convert it from Caffe to Tensorflow?
@twhui I added link to your license to readme. Also implemented separate converter for weights directly from Caffe to Tensorflow.
@keeper121 A link to your repo is added.