diff options
Diffstat (limited to 'Code/g_model.py')
| -rw-r--r-- | Code/g_model.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/Code/g_model.py b/Code/g_model.py index 4e0df31..5dc8265 100644 --- a/Code/g_model.py +++ b/Code/g_model.py @@ -106,15 +106,14 @@ class GeneratorModel: scale_height = int(height * scale_factor) scale_width = int(width * scale_factor) - inputs = tf.image.resize_images(inputs, scale_height, scale_width) - scale_gts = tf.image.resize_images(gts, scale_height, scale_width) + inputs = tf.image.resize_images(inputs, [scale_height, scale_width]) + scale_gts = tf.image.resize_images(gts, [scale_height, scale_width]) # for all scales but the first, add the frame generated by the last # scale to the input if scale_num > 0: - last_gen_frames = tf.image.resize_images(last_gen_frames, - scale_height, - scale_width) + last_gen_frames = tf.image.resize_images( + last_gen_frames,[scale_height, scale_width]) inputs = tf.concat(3, [inputs, last_gen_frames]) # generated frame predictions |