search

hmorimitsu / ptlflow

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.
Apache License 2.0
250 stars 33 forks source link

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 #33

Closed izhimahu closed 2 years ago

hmorimitsu commented 2 years ago

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?

izhimahu commented 2 years ago

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?

image These are caffe models, is there a pytorch trained model?

hmorimitsu commented 2 years ago

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.