diff options
Diffstat (limited to 'live-mogrify.py')
| -rw-r--r-- | live-mogrify.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index 5cd7f67..9fba009 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -133,13 +133,13 @@ def process_image(opt, data_opt, im): def list_checkpoints(payload): print("> list checkpoints") - return sorted([f.split('/')[2] for f in glob.glob('./checkpoints/' + payload + '/*/latest_net_G.pth')]) + return sorted([f.split('/')[3] for f in glob.glob('./checkpoints/' + payload + '/*/latest_net_G.pth')]) def list_epochs(path): print("> list epochs for {}".format(path)) if not os.path.exists(os.path.join('./checkpoints/', path)): return "not found" - return sorted([f.split('/')[3].replace('_net_G.pth','') for f in glob.glob('./checkpoints/' + path + '/*_net_G.pth')]) + return sorted([f.split('/')[4].replace('_net_G.pth','') for f in glob.glob('./checkpoints/' + path + '/*_net_G.pth')]) def list_sequences(module): print("> list sequences") |