lmb-freiburg / flownet2

FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
https://lmb.informatik.uni-freiburg.de/Publications/2017/IMKDB17/
Other
1k stars 318 forks source link

how to change the batch size when testing #123

Closed jeremy43 closed 6 years ago

jeremy43 commented 6 years ago

Hi, when I try to run run-flownet-many, the batch size is one pair img (img0, img1,.flo), and took about 1s per pair. Ss there any way to change to batch size when testing? Thanks!

nikolausmayer commented 6 years ago

The input blobs are assembled here. You can change that code to stack multiple image pairs into one batch. Note that you'll also need to change the output code accordingly.

If you are concerned about speed, you might want to try what https://github.com/lmb-freiburg/flownet2/issues/105 suggested first.

nikolausmayer commented 6 years ago

(closed due to inactivity)