1 files changed, 3 insertions, 3 deletions
diff --git a/main.py b/main.py
index 9fceccf1..4497cce0 100644
--- a/main.py
+++ b/main.py
@@ -13,6 +13,7 @@ from losses import YourModel
 from skimage.io import imread, imsave
 from matplotlib import pyplot as plt
 import numpy as np
+from skimage import transform
 
 
 def parse_args():
@@ -61,8 +62,7 @@ def main():
 
     content_image = imread(ARGS.content)
     style_image = imread(ARGS.style)
-    style_image = transform.resize(style_image, content_image.shape, anti_aliasing=True)
-    
+    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)
@@ -70,7 +70,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)