From 1d8b79f61492cbb28ce0f087f3a9dcea8546e06b Mon Sep 17 00:00:00 2001 From: quillotm Date: Thu, 2 Sep 2021 17:04:07 +0200 Subject: [PATCH] Changed to make simple the utt2sub and sub2utt commands. Was too long, no it cost less time. --- volia/masseffect.py | 34 +++++++++++++++------------------- 1 file changed, 15 insertions(+), 19 deletions(-) diff --git a/volia/masseffect.py b/volia/masseffect.py index 8dc1467..0163eea 100644 --- a/volia/masseffect.py +++ b/volia/masseffect.py @@ -92,32 +92,30 @@ def converter(file: str, outtype: str, outfile: str): of.write(f"{kaldi_id} {masseffect_id}\n") -def utt2sub(file: str, subfile: str, outfile: str): +def utt2sub(file: str, outfile: str): data = core.data.read_id_values(file) keys = [key for key in data] - data_sub = core.data.read_id_values(subfile) - keys_sub = [key for key in data_sub] - with open(outfile, "w") as of: - for key in keys: - subkeys = [subkey for subkey in keys_sub if subkey.startswith(key)] - subkeys_str = " ".join(subkeys) + key_2_subkeys = {} + for subkey in keys: + key = subkey.replace(" ", "").replace("\n", "").split("_")[:-1] + key_2_subkeys[key] = [] + key_2_subkeys[key].append(subkey) + + for key in key_2_subkeys: + subkeys_str = " ".join(key_2_subkeys[key]) of.write(f"{key} {subkeys_str}\n") -def sub2utt(file: str, subfile: str, outfile: str): +def sub2utt(file: str, outfile: str): data = core.data.read_id_values(file) keys = [key for key in data] - data_sub = core.data.read_id_values(subfile) - keys_sub = [key for key in data_sub] - with open(outfile, "w") as of: - for key in keys: - subkeys = [subkey for subkey in keys_sub if subkey.startswith(key)] - for subkey in subkeys: - of.write(f"{subkey} {key}\n") + for subkey in keys: + key = subkey.replace(" ", "").replace("\n", "").split("_")[:-1] + of.write(f"{subkey} {key}\n") if __name__ == '__main__': @@ -162,15 +160,13 @@ if __name__ == '__main__': # Create utt2sub parser_utt2sub = subparsers.add_parser("utt2sub", help="generate utt2sub file") - parser_utt2sub.add_argument("--file", required=True, type=str, help="features, list or labels file with normal ids") - parser_utt2sub.add_argument("--subfile", required=True, type=str, help="features, list or labels file with sub ids") + parser_utt2sub.add_argument("--file", required=True, type=str, help="features, list or labels file with sub ids") parser_utt2sub.add_argument("--outfile", required=True, type=str, help="output file") parser_utt2sub.set_defaults(which="utt2sub") # Create sub2utt parser_sub2utt = subparsers.add_parser("sub2utt", help="generate sub2utt file") - parser_sub2utt.add_argument("--file", required=True, type=str, help="features, list or labels file with normal ids") - parser_sub2utt.add_argument("--subfile", required=True, type=str, help="features, list or labels file sub ids") + parser_sub2utt.add_argument("--file", required=True, type=str, help="features, list or labels file with sub ids") parser_sub2utt.add_argument("--outfile", required=True, type=str, help="output file") parser_sub2utt.set_defaults(which="sub2utt") -- 1.8.2.3