From a438c6e88484a68abdd17384e720452ffa2b96bb Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 7 Jun 2018 01:08:51 +0200 Subject: module name --- live-mogrify.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'live-mogrify.py') diff --git a/live-mogrify.py b/live-mogrify.py index 6e62645..89ce10c 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -38,6 +38,8 @@ def load_opt(): opt = opt_parser.parse() data_opt_parser = DatasetOptions() data_opt = data_opt_parser.parse(opt.unknown) + global module_name + module_name = data_opt.module_name opt.nThreads = 1 # test code only supports nThreads = 1 opt.batchSize = 1 # test code only supports batchSize = 1 opt.serial_batches = True # no shuffle @@ -141,10 +143,10 @@ def list_epochs(path): def list_sequences(): print("> list sequences") - sequences = sorted([name for name in os.listdir('./sequences') if os.path.isdir(os.path.join('./sequences/', name))]) + sequences = sorted([name for name in os.listdir(os.path.join('./sequences/', module_name)) if os.path.isdir(os.path.join('./sequences/', module_name, name))]) results = [] for path in sequences: - count = len([name for name in os.listdir(os.path.join('./sequences/', path)) if os.path.isfile(os.path.join('./sequences/', path, name))]) + count = len([name for name in os.listdir(os.path.join('./sequences/', module_name, path)) if os.path.isfile(os.path.join('./sequences/', module_name, path, name))]) results.append({ 'name': path, 'count': count, @@ -153,7 +155,7 @@ def list_sequences(): def read_sequence(path): print("> read sequence {}".format(path)) - return sorted([f for f in glob.glob(os.path.join('./sequences/', path, '*.png'))]) + return sorted([f for f in glob.glob(os.path.join('./sequences/', module_name, path, '*.png'))]) class Listener(): def __init__(self): -- cgit v1.2.3-70-g09d2