Spaces:
Sleeping
Sleeping
| #!/usr/bin/env python3 | |
| import argparse | |
| from typing import Type | |
| from espnet2.main_funcs.pack_funcs import pack | |
| class PackedContents: | |
| files = [] | |
| yaml_files = [] | |
| class ASRPackedContents(PackedContents): | |
| # These names must be consistent with the argument of inference functions | |
| files = ["asr_model_file", "lm_file"] | |
| yaml_files = ["asr_train_config", "lm_train_config"] | |
| class TTSPackedContents(PackedContents): | |
| files = ["model_file"] | |
| yaml_files = ["train_config"] | |
| class EnhPackedContents(PackedContents): | |
| files = ["model_file"] | |
| yaml_files = ["train_config"] | |
| class DiarPackedContents(PackedContents): | |
| files = ["model_file"] | |
| yaml_files = ["train_config"] | |
| def add_arguments(parser: argparse.ArgumentParser, contents: Type[PackedContents]): | |
| parser.add_argument("--outpath", type=str, required=True) | |
| for key in contents.yaml_files: | |
| parser.add_argument(f"--{key}", type=str, default=None) | |
| for key in contents.files: | |
| parser.add_argument(f"--{key}", type=str, default=None) | |
| parser.add_argument("--option", type=str, action="append", default=[]) | |
| def get_parser() -> argparse.ArgumentParser: | |
| parser = argparse.ArgumentParser(description="Pack input files to archive format") | |
| subparsers = parser.add_subparsers() | |
| # Create subparser for ASR | |
| for name, contents in [ | |
| ("asr", ASRPackedContents), | |
| ("tts", TTSPackedContents), | |
| ("enh", EnhPackedContents), | |
| ("diar", DiarPackedContents), | |
| ]: | |
| parser_asr = subparsers.add_parser( | |
| name, | |
| formatter_class=argparse.ArgumentDefaultsHelpFormatter, | |
| ) | |
| add_arguments(parser_asr, contents) | |
| parser_asr.set_defaults(contents=contents) | |
| return parser | |
| def main(cmd=None): | |
| parser = get_parser() | |
| args = parser.parse_args(cmd) | |
| if not hasattr(args, "contents"): | |
| parser.print_help() | |
| parser.exit(2) | |
| yaml_files = { | |
| y: getattr(args, y) | |
| for y in args.contents.yaml_files | |
| if getattr(args, y) is not None | |
| } | |
| files = { | |
| y: getattr(args, y) for y in args.contents.files if getattr(args, y) is not None | |
| } | |
| pack( | |
| yaml_files=yaml_files, | |
| files=files, | |
| option=args.option, | |
| outpath=args.outpath, | |
| ) | |
| if __name__ == "__main__": | |
| main() | |