djl11 / pwcnettensorflow Goto Github PK

Thank you for making TensorFlow Version of PWC_Net.
Can you share your trained weight?
Also got the error when running main.py

Hi @djl11,
Thanks for sharing your code.

The optical flow estimator network is a feed-forward network formed by stacking layers one over the other:

In the flow_estimator_forward_pass function, where the estimator network is defined, it says:

conv4 = tf.layers.conv2d(conv3,64,[3,3],(1,1),'same','channels_first',(1,1),tf.nn.leaky_relu)
ft = tf.layers.conv2d(conv4,32,[3,3],(1,1),'same','channels_first',(1,1),tf.nn.leaky_relu)
wt = tf.layers.conv2d(conv4,2,[3,3],(1,1),'same','channels_first',(1,1),tf.nn.leaky_relu)

That is, both ft and wt are stacked over conv4, instead of being stacked one over the other as in the figure from the paper. Was this intentional, or a bug?

hi @djl11
Thanks for your work. I want to know if I want to fine-tune on the Kitti Dataset, the ground truth flow are sparse. so how can I change the Loss Function.