X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=voxel_flow_model.py;fp=voxel_flow_model.py;h=b3f554d59f935ef327a8a1b86d936461988cb779;hb=c74057bc77fb9bc4eb75668896adfb85ddb99251;hp=d419acf6a4493cd534c881c60fbee73bd2e57bff;hpb=586b7753362bf4b02fec6490dcefe4073f1af8bd;p=voxel-flow diff --git a/voxel_flow_model.py b/voxel_flow_model.py index d419acf..b3f554d 100755 --- a/voxel_flow_model.py +++ b/voxel_flow_model.py @@ -85,6 +85,6 @@ class Voxel_flow_model(object): mask = 0.5 * (1.0 + mask) mask = tf.tile(mask, [1, 1, 1, 3]) - net = tf.mul(mask, output_1) + tf.mul(1.0 - mask, output_2) + net = tf.multiply(mask, output_1) + tf.multiply(1.0 - mask, output_2) return net