diff options
| author | David Doan <daviddoan@davids-mbp-3.devices.brown.edu> | 2022-05-07 17:41:40 -0400 |
|---|---|---|
| committer | David Doan <daviddoan@davids-mbp-3.devices.brown.edu> | 2022-05-07 17:41:40 -0400 |
| commit | 423c72c1eb93ea7fc6aaeba29e1c9f5e376b9fe6 (patch) | |
| tree | f3994f63e3c2a476ed3cd31bbec168d2a6520557 /main.py | |
| parent | 228d83a47b6c6eb1d0c01b85761a8f6e1db48c1d (diff) | |
testing
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -54,7 +54,7 @@ def main(): content_image = imread(ARGS.content) style_image = imread(ARGS.style) - style_image = transform.resize(style_image, content_image.shape).astype('uint8') + style_image = transform.resize(style_image, content_image.shape) my_model = YourModel(content_image=content_image, style_image=style_image) my_model.vgg16.build([1, 255, 255, 3]) my_model.vgg16.load_weights('vgg16_imagenet.h5', by_name=True) @@ -62,7 +62,7 @@ def main(): final_image = tf.squeeze(my_model.x) - plt.imshow(final_image) + plt.imshow(final_image).astype('uint8') imsave(ARGS.savefile, final_image) |