NVIDIA / flownet2-pytorch

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

inference error #267

Open poincarelee opened 2 years ago

poincarelee commented 2 years ago

Hi, I have run the inference script, in the main script batchsize in inference dataset is 4, but when saving flow result, the batchsize is 1, which means input 4 pics and get one .flo file, actually the saving flo file is the first result, the other 3 flo result are discarded?