aboutsummaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
authorLogan Bauman <logan_bauman@brown.edu>2022-05-07 14:26:42 -0400
committerLogan Bauman <logan_bauman@brown.edu>2022-05-07 14:26:42 -0400
commitaa4999f22143be058cb73c829783bf5f894c7c0f (patch)
treebae64f9c1957daeed3486cbdc8fb14943109b0fe /main.py
parent1b54bfa9de44f29ca2046d1eeb0ab174ae0dadbd (diff)
hi
Diffstat (limited to 'main.py')
-rw-r--r--main.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/main.py b/main.py
index 60aa6b07..45714611 100644
--- a/main.py
+++ b/main.py
@@ -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])