| |
| |
| |
| |
|
|
| import re |
| from preprocessors import ( |
| m4singer, |
| opencpop, |
| svcc, |
| pjs, |
| popbutfy, |
| opensinger, |
| popcs, |
| kising, |
| csd, |
| opera, |
| nus48e, |
| svcceval, |
| vctk, |
| vctksample, |
| libritts, |
| lijian, |
| cdmusiceval, |
| ljspeech, |
| coco, |
| cocoeval, |
| custom, |
| vocalist, |
| ljspeech_vocoder, |
| ) |
|
|
|
|
| def preprocess_dataset( |
| dataset, dataset_path, output_path, cfg, is_custom_dataset=False |
| ): |
| """Call specific function to handle specific dataset |
| Args: |
| dataset (str): name of a dataset, e.g. opencpop, m4singer |
| dataset_path (str): path to dataset |
| output_path (str): path to store preprocessing result files |
| """ |
| if is_custom_dataset: |
| custom.main(output_path, dataset_path, dataset_name=dataset) |
| return |
|
|
| if re.match("opencpop*", dataset): |
| opencpop.main(dataset, output_path, dataset_path) |
| if dataset == "m4singer": |
| m4singer.main(output_path, dataset_path) |
| if dataset == "svcc": |
| svcc.main(output_path, dataset_path) |
| if dataset == "pjs": |
| pjs.main(output_path, dataset_path) |
| if dataset == "popbutfy": |
| popbutfy.main(output_path, dataset_path) |
| if dataset == "opensinger": |
| opensinger.main(output_path, dataset_path) |
| if dataset == "popcs": |
| popcs.main(output_path, dataset_path) |
| if dataset == "kising": |
| kising.main(output_path, dataset_path) |
| if dataset == "csd": |
| csd.main(output_path, dataset_path) |
| if dataset == "opera": |
| opera.main(output_path, dataset_path) |
| if dataset == "nus48e": |
| nus48e.main(output_path, dataset_path) |
| if dataset == "vctk": |
| vctk.main(output_path, dataset_path) |
| if dataset == "svcceval": |
| svcceval.main(output_path, dataset_path) |
| if dataset == "libritts": |
| libritts.main(output_path, dataset_path) |
| if dataset == "lijian": |
| lijian.main(output_path, dataset_path) |
| if dataset == "cdmusiceval": |
| cdmusiceval.main(output_path, dataset_path) |
| if dataset == "LJSpeech": |
| ljspeech.main(output_path, dataset_path, cfg) |
| if dataset == "ljspeech": |
| ljspeech_vocoder.main(output_path, dataset_path) |
| if dataset == "coco": |
| coco.main(output_path, dataset_path) |
| if dataset == "cocoeval": |
| cocoeval.main(output_path, dataset_path) |
| if dataset == "vocalist": |
| vocalist.main(output_path, dataset_path) |
|
|
|
|
| def prepare_align(dataset, dataset_path, cfg, output_path): |
| """Call specific function to handle specific dataset |
| |
| Args: |
| dataset (str): name of a dataset, e.g. ljspeech |
| dataset_path (str): path to dataset |
| output_path (str): path to store preprocessing result files |
| """ |
| if dataset == "LJSpeech": |
| ljspeech.prepare_align(dataset, dataset_path, cfg, output_path) |
|
|
