Spaces:
Running
Running
updated neural style transfer
Browse files- 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('
|
| 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
|