diff options
-rw-r--r-- | main.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -53,7 +53,7 @@ def main(): content_image = imread(ARGS.content) style_image = imread(ARGS.style) - style_image = transform.resize(style_image, content_image.shape) + style_image = transform.resize(style_image, content_image.shape, anti_aliasing=True) my_model = YourModel(content_image=content_image, style_image=style_image) my_model.vgg16.build([1, 255, 255, 3]) |