changed out_path
Browse files
test.py
CHANGED
|
@@ -148,7 +148,7 @@ def main():
|
|
| 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
|
| 152 |
save_image(out_tensor, out_pth)
|
| 153 |
|
| 154 |
if args.grid_pth:
|
|
|
|
| 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:
|