add two categories for MASA Mushra tests (1baedaf7) · Commits · IVAS Codec Public Collaboration / IVAS Processing Scripts

experiments/create_experiment_config.py

+11 −1

Original line number	Diff line number	Diff line
		@@ -13,6 +13,8 @@ HERE = Path(__file__).parent.absolute().resolve()
		IN_FMT_FOR_MASA_EXPS = {
		"P800-8": dict(zip([f"cat{i}" for i in range(1, 7)], ["FOA"] * 6)),
		"P800-9": dict(zip([f"cat{i}" for i in range(1, 7)], ["FOA"] * 6)),
		"BS1534-7a": dict(zip([f"cat{i}" for i in range(1, 3)], ["FOA"] * 2)),
		"BS1534-7b": dict(zip([f"cat{i}" for i in range(1, 3)], ["FOA"] * 2)),
		}


		@@ -23,7 +25,15 @@ def _get_seed(exp, lab):
		def create_experiment_setup(experiment, lab) -> list[Path]:
		default_cfg_path = HERE.joinpath(f"selection/{experiment}/config/{experiment}.yml")

		categories = [f"cat{i}" for i in range(1, 7)] if experiment in EXPERIMENTS_P800 else [""]
		num_categories = 1
		# for P800 we nee 6 categories (for each background/scene)
		if experiment in EXPERIMENTS_P800:
		num_categories = 6
		# for MUSHRA, we only need categories in the MASA tests, as there could be both FOA and HOA2 input
		elif experiment in IN_FMT_FOR_MASA_EXPS:
		num_categories = 2
		categories = [f"cat{i}" for i in range(1, num_categories + 1)]

		seed = _get_seed(experiment, lab)
		base_path = Path(HERE.name).joinpath(f"selection/{experiment}")