Fix yet more no-movement rejection (we would reject A-A-B pulldown, but not A-B-B).

[voxel-flow] / voxel_flow_model.py

diff --git a/voxel_flow_model.py b/voxel_flow_model.py
index 2cca42054555f85e98998ca7dc67d339e77ad456..6193e612f89e94ba89be9324f79ecdbd7e83d90b 100755 (executable)
--- a/voxel_flow_model.py
+++ b/voxel_flow_model.py
@@ -66,6 +66,7 @@ class Voxel_flow_model(object):
           net = slim.conv2d(net, 64, [5, 5], stride=1, scope='conv6')
     net = slim.conv2d(net, 3, [5, 5], stride=1, activation_fn=tf.tanh,
     normalizer_fn=None, scope='conv7')
+    net_copy = net
     
     flow = net[:, :, :, 0:2]
     mask = tf.expand_dims(net[:, :, :, 2], 3)
@@ -89,4 +90,4 @@ class Voxel_flow_model(object):
     mask = tf.tile(mask, [1, 1, 1, 3])
     net = tf.multiply(mask, output_1) + tf.multiply(1.0 - mask, output_2)
 
-    return net
+    return [net, net_copy]