summaryrefslogtreecommitdiff
path: root/live-mogrify.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-21 15:28:42 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-21 15:28:42 +0200
commitd7d14946e13afc1e921ca31f8d54fab0e3f5253d (patch)
tree88040669faf12ae32a28cda269d07ebe78bf73e6 /live-mogrify.py
parent6d10b21c010b0f7060c051b94d9a97c29ec400da (diff)
cmdz
Diffstat (limited to 'live-mogrify.py')
-rw-r--r--live-mogrify.py3
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...']