ClementPinard / FlowNetPytorch

Pytorch implementation of FlowNet by Dosovitskiy et al.
MIT License
842 stars 206 forks source link

PreTrained FlowNetS with BatchNormalization #12

Closed sparviero19 closed 6 years ago

sparviero19 commented 6 years ago

Hi,

is there a pretrained pytorch model of the batch normalized version of FlownetS? Thanks

ClementPinard commented 6 years ago

Pretrained models will come when I manage to get ~3 EPE. Still not reaching it, my best so far is 3.9 Stay tuned !

taineleau-zz commented 6 years ago

How long does it take to achieve 3.9? Could you also share me with the server config?

ClementPinard commented 6 years ago

Hello, sorry for the late answer, convergence problems are now solved, documentation and networks can now be found here

Clément