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)
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]
# Prepare model.
model = Voxel_flow_model()
- prediction = model.inference(input_placeholder)
+ prediction, flow = model.inference(input_placeholder)
# reproduction_loss, prior_loss = model.loss(prediction, target_placeholder)
reproduction_loss = model.loss(prediction, target_placeholder)
# total_loss = reproduction_loss + prior_loss
summaries.append(tf.summary.image('Input Image (after)', input_placeholder[:, :, :, 3:6], 3));
summaries.append(tf.summary.image('Output Image', prediction, 3))
summaries.append(tf.summary.image('Target Image', target_placeholder, 3))
+ summaries.append(tf.summary.image('Flow', flow, 3))
# Create a saver.
saver = tf.train.Saver(tf.all_variables())
# target_resized = tf.image.resize_area(target_placeholder,[128, 128])
# Prepare model.
- model = Voxel_flow_model(is_train=True)
+ model, flow = Voxel_flow_model(is_train=True)
prediction = model.inference(input_placeholder)
# reproduction_loss, prior_loss = model.loss(prediction, target_placeholder)
reproduction_loss = model.loss(prediction, target_placeholder)