diff --git a/ivas_processing_scripts/generation/__init__.py b/ivas_processing_scripts/generation/__init__.py index a07bd7331f193ffed3744d983257cee1dc1a42f2..8ee7c026d64269a0de0d7988b9df7ea5424a66e4 100755 --- a/ivas_processing_scripts/generation/__init__.py +++ b/ivas_processing_scripts/generation/__init__.py @@ -80,9 +80,9 @@ def main(args): # parse configuration cfg = config.TestConfig(args.config) - # create output directory, if not existing - if not os.path.exists(cfg.output_path): - create_dir(cfg.output_path) + # create output directories for categories + for cat in range(1, 7): + create_dir(cfg.output_path.joinpath(f"cat{cat}")) # set up logging logger = logging_init(args, cfg)