diff options
| -rw-r--r-- | live-mogrify.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index a709bdc..4e535e3 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -126,12 +126,15 @@ def process_image(opt, data_opt, im): return img def list_checkpoints(): + print("> list checkpoints") return glob.glob('./checkpoints/*/latest_net_G*') def list_checkpoint_dir(path): + print("> list checkpoint path {}".format(path)) if not os.path.exists('./checkpoints/' + path): return "not found" return glob.glob(os.path.join('./checkpoints/' + path + '/*_net_G.pth')) def list_datasets(): + print("> list datasets") # return glob.glob('./checkpoints/*/latest_net_G*') return ['coming soon...'] |