Is there a pytorch pre-training model for LiteFlowNet? I found it very memory intensive when I applied draft, so I need a light weight optical flow estimation network
izhimahu opened this issue · 3 comments
I am sorry, I am not sure I understand your problem. LiteFlowNet and its weights are provided in this repository. Is this what you need, or do you mean something different?
I am sorry, I am not sure I understand your problem. LiteFlowNet and its weights are provided in this repository. Is this what you need, or do you mean something different?
These are caffe models, is there a pytorch trained model?
Maybe it is a bit confusing, but this name in the second column is the name of the original checkpoint that I used to generate my own version (the name in the first column). The original LiteFlownet was released in Caffe, but I converted it into Pytorch. Therefore, the checkpoint that you will get when using PTLFlow will be the Pytorch version (liteflownet3-....ckpt).
The provided scripts can download the weights when you select to use a pretrained model. But if you want to download only the weights alone by yourself, you can find them here: https://github.com/hmorimitsu/ptlflow/releases/tag/weights1
I hope that helps you. If not, just let me know.