TungDuong commited on
Commit
24597f6
·
verified ·
1 Parent(s): 0b3ae2a

updated neural style transfer

Browse files
Files changed (1) hide show
  1. NeuralStyleTransfer.py +1 -2
NeuralStyleTransfer.py CHANGED
@@ -27,7 +27,7 @@ def load_json_config(file_path):
27
  config = json.load(file)
28
  return config
29
 
30
- config = load_json_config('../NeuralStyleTransfer/config.json')
31
 
32
  def load_image(image_link):
33
  image = cv2.imread(image_link)
@@ -112,5 +112,4 @@ class NeuralStyleTransfer(nn.Module):
112
 
113
  styled_image = convert_image_origin_size(generated_image, original_height, original_width)
114
 
115
- vutils.save_image(generated_image, 'results/generated_image.jpg', normalize=True)
116
  return styled_image
 
27
  config = json.load(file)
28
  return config
29
 
30
+ config = load_json_config('config.json')
31
 
32
  def load_image(image_link):
33
  image = cv2.imread(image_link)
 
112
 
113
  styled_image = convert_image_origin_size(generated_image, original_height, original_width)
114
 
 
115
  return styled_image