s486797/neural_style
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

modify mode_style_transfer code

Browse Source
This commit is contained in:
s486797 2024-10-20 21:02:21 +02:00
parent 6b9ae538fd
commit 909824d0ed
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
neural_style_app/mode_style_transfer.py
View File

@ -157,7 +157,6 @@ class StyleTransferModel:
def __init__(self, content_img, style_img, num_steps=300, style_weight=1000000, content_weight=1): def __init__(self, content_img, style_img, num_steps=300, style_weight=1000000, content_weight=1):
self.content_img = content_img self.content_img = content_img
self.style_img = style_img.resize(content_img.size) 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.style_img = image_loader(self.style_img)
self.content_img = image_loader(self.content_img) self.content_img = image_loader(self.content_img)
self.input_img = self.content_img.clone() self.input_img = self.content_img.clone()