NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
Other
3.09k stars 739 forks source link

How to get horizontal and vertical optical flow? #221

Closed Senwang98 closed 3 years ago

Senwang98 commented 3 years ago

Hi, To reproduction one paper, i need to use optical flow. However, that paper uses horizontal and vertical optical flow(X-Y) rather than one optical flow, my question is that can FlowNet2 supports to get horizontal and vertical optical flow? Thanks.

albertchristian92 commented 3 years ago

Hi @1187697147 did you solve how to find horizontal and vertical optical flow?

Thank you

Senwang98 commented 3 years ago

Find the line that saves the flo file in main.py There is an intermediate variable in the upper line of save flo, you can split that variable into two directions and save it.