| | |
| | |
| | |
| | |
| |
|
| | 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) |
| |
|
