From 909824d0edde61ec9b62011644ad980d05302470 Mon Sep 17 00:00:00 2001 From: s486797 Date: Sun, 20 Oct 2024 21:02:21 +0200 Subject: [PATCH] modify mode_style_transfer code --- neural_style_app/mode_style_transfer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/neural_style_app/mode_style_transfer.py b/neural_style_app/mode_style_transfer.py index fadfecf..b8c0705 100644 --- a/neural_style_app/mode_style_transfer.py +++ b/neural_style_app/mode_style_transfer.py @@ -157,7 +157,6 @@ class StyleTransferModel: def __init__(self, content_img, style_img, num_steps=300, style_weight=1000000, content_weight=1): self.content_img = content_img self.style_img = style_img.resize(content_img.size) - #self.style_img = self.style_img.resize(self.content_img.size) self.style_img = image_loader(self.style_img) self.content_img = image_loader(self.content_img) self.input_img = self.content_img.clone()