def warp_1c(I, F, H, W):
return tf.reshape(
dense_image_warp(I, tf.stack([-F[..., 1], -F[..., 0]], -1)),
[FLAGS.batch_size, H, W, 1])
def warp(self, I, F, b, h, w, c):
return tf.reshape(
dense_image_warp(I, tf.stack([-F[..., 1], -F[..., 0]], -1)),
[b, h, w, c])
def warp(I, F):
return tf.reshape(
dense_image_warp(I, tf.stack([-F[..., 1], -F[..., 0]], -1)),
[FLAGS.batch_size, CROP_PATCH_H, CROP_PATCH_W, 3])
def warp_with_large_size(I, F, c):
return tf.reshape(dense_image_warp(I, tf.stack([-F[..., 1], -F[..., 0]], -1)), [FLAGS.batch_size, RESIZED_H, RESIZED_W, c])
