Sophie98 commited on
Commit
40845c9
1 Parent(s): e4fb230
Files changed (1) hide show
  1. styleTransfer.py +2 -2
styleTransfer.py CHANGED
@@ -34,7 +34,7 @@ def content_transform():
34
  transform = transforms.Compose(transform_list)
35
  return transform
36
 
37
- def Transformer(content_img: Image, style_img: Image,
38
  vgg_path:str = 'vgg_normalised.pth', decoder_path:str = 'decoder_iter_160000.pth',
39
  Trans_path:str = 'transformer_iter_160000.pth', embedding_path:str = 'embedding_iter_160000.pth'):
40
  # Advanced options
@@ -108,7 +108,7 @@ def Transformer(content_img: Image, style_img: Image,
108
  return output
109
 
110
  def create_styledSofa(sofa:Image, style:Image):
111
- styled_sofa = transformer(sofa,style)
112
  return styled_sofa
113
 
114
  # image = Image.open('sofa_office.jpg')
 
34
  transform = transforms.Compose(transform_list)
35
  return transform
36
 
37
+ def StyleTransformer(content_img: Image, style_img: Image,
38
  vgg_path:str = 'vgg_normalised.pth', decoder_path:str = 'decoder_iter_160000.pth',
39
  Trans_path:str = 'transformer_iter_160000.pth', embedding_path:str = 'embedding_iter_160000.pth'):
40
  # Advanced options
 
108
  return output
109
 
110
  def create_styledSofa(sofa:Image, style:Image):
111
+ styled_sofa = StyleTransformer(sofa,style)
112
  return styled_sofa
113
 
114
  # image = Image.open('sofa_office.jpg')