changed out_path
Browse files- __pycache__/AdaIN.cpython-39.pyc +0 -0
- __pycache__/Network.cpython-39.pyc +0 -0
- __pycache__/utils.cpython-39.pyc +0 -0
- test.py +2 -0
__pycache__/AdaIN.cpython-39.pyc
ADDED
|
Binary file (1.87 kB). View file
|
|
|
__pycache__/Network.cpython-39.pyc
ADDED
|
Binary file (2.13 kB). View file
|
|
|
__pycache__/utils.cpython-39.pyc
ADDED
|
Binary file (4.31 kB). View file
|
|
|
test.py
CHANGED
|
@@ -147,6 +147,8 @@ def main():
|
|
| 147 |
|
| 148 |
# Save image
|
| 149 |
out_pth = out_dir + content_pth.stem + '_style_' + style_pth.stem + '_alpha' + str(args.alpha) + content_pth.suffix
|
|
|
|
|
|
|
| 150 |
save_image(out_tensor, out_pth)
|
| 151 |
|
| 152 |
if args.grid_pth:
|
|
|
|
| 147 |
|
| 148 |
# Save image
|
| 149 |
out_pth = out_dir + content_pth.stem + '_style_' + style_pth.stem + '_alpha' + str(args.alpha) + content_pth.suffix
|
| 150 |
+
if args.color_control:
|
| 151 |
+
out_pth + '_colorcontrol'
|
| 152 |
save_image(out_tensor, out_pth)
|
| 153 |
|
| 154 |
if args.grid_pth:
|