diff options
| author | Vee9ahd1 <[email protected]> | 2019-05-20 23:24:07 -0400 |
|---|---|---|
| committer | Vee9ahd1 <[email protected]> | 2019-05-20 23:24:07 -0400 |
| commit | dcc12c31b79ac9a17738d75adba65556624c412e (patch) | |
| tree | 94ca1c05596e9f4a60441e819d519b0c42db523f /gantools/cli.py | |
| parent | 9a67c02438e1e3231340b5019dd9fb1b43a4f779 (diff) | |
prefix bug fix
Diffstat (limited to 'gantools/cli.py')
| -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.') |