diff options
| -rw-r--r-- | gantools/cli.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/gantools/cli.py b/gantools/cli.py index d15e01d..73213ed 100644 --- a/gantools/cli.py +++ b/gantools/cli.py @@ -38,7 +38,8 @@ def main(): ims = gan.sample(z_seq, label_seq, truncation_seq, args.nbatch) # save images to file - print('Saving image files: '+str(args.pathprefix)) - image_utils.save_images(ims, args.pathprefix) + pathprefix = '' if args.pathprefix == None else str(args.pathprefix) + print('Saving image files: '+pathprefix) + image_utils.save_images(ims, pathprefix) print('Done.') |