lmb-freiburg / flownet2

FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Home Page:https://lmb.informatik.uni-freiburg.de/Publications/2017/IMKDB17/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

train.prototxt for flying chairs

kindloaf opened this issue · comments

I'm trying to train flownet_S using flying chairs data.
The doc says train.prototxt should be created based on train.prototxt.template, and "make sure the correct parts of the network are enabled by setting/adding loss weights and blob learning rates."

I did the following changes, do they look good / complete?

$ diff FlowNet2-S_train.prototxt.template train.prototxt
14c14
<     source: "../../data/FlyingThings3D_release_TRAIN_lmdb"
---
>     source: "../../data/FlyingChairs_release_lmdb"
43c43
<     source: "../../data/FlyingThings3D_release_TEST_lmdb"
---
>     source: "../../data/FlyingChairs_release_lmdb"
214,215c214,215
<     crop_width: 768
<     crop_height: 384
---
>     crop_width: 448
>     crop_height: 320
310,311c310,311
<     crop_width: 768
<     crop_height: 384
---
>     crop_width: 448
>     crop_height: 320
331,332c331,332
<     crop_width: 768
<     crop_height: 384
---
>     crop_width: 448
>     crop_height: 320

I think that should work for an "S" net on FlyingChairs, yes.

Cool. Thanks.