lmb-freiburg / flownet2-docker

Dockerfile and runscripts for FlowNet 2.0 (estimation of optical flow)
https://lmb.informatik.uni-freiburg.de/Publications/2017/IMKDB17/
GNU General Public License v3.0
158 stars 52 forks source link

variable input sizes #12

Closed dvdhfnr closed 3 years ago

dvdhfnr commented 5 years ago

To allow variable input sizes, Line 92 in FN2_run-flownet-docker.py should be exchanged by sth like

if width != input_data[0].shape[3] or height != input_data[0].shape[2]:

where width and height might be initialised with -1.