| import os | |
| import json | |
| import pathlib | |
| from random import shuffle | |
| from rvc.configs.config import Config | |
| config = Config() | |
| current_directory = os.getcwd() | |
| def generate_config(rvc_version, sampling_rate, model_path): | |
| if rvc_version == "v1" or sampling_rate == "40000": | |
| config_path = f"v1/{sampling_rate}.json" | |
| else: | |
| config_path = f"v2/{sampling_rate}.json" | |
| config_save_path = os.path.join(model_path, "config.json") | |
| if not pathlib.Path(config_save_path).exists(): | |
| with open(config_save_path, "w", encoding="utf-8") as f: | |
| json.dump( | |
| config.json_config[config_path], | |
| f, | |
| ensure_ascii=False, | |
| indent=4, | |
| sort_keys=True, | |
| ) | |
| f.write("\n") | |
| def generate_filelist(f0_method, model_path, rvc_version, sampling_rate): | |
| gt_wavs_dir = f"{model_path}/0_gt_wavs" | |
| feature_dir = ( | |
| f"{model_path}/3_feature256" | |
| if rvc_version == "v1" | |
| else f"{model_path}/3_feature768" | |
| ) | |
| if f0_method: | |
| f0_dir = f"{model_path}/2a_f0" | |
| f0nsf_dir = f"{model_path}/2b-f0nsf" | |
| names = ( | |
| set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)]) | |
| & set([name.split(".")[0] for name in os.listdir(feature_dir)]) | |
| & set([name.split(".")[0] for name in os.listdir(f0_dir)]) | |
| & set([name.split(".")[0] for name in os.listdir(f0nsf_dir)]) | |
| ) | |
| else: | |
| names = set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)]) & set( | |
| [name.split(".")[0] for name in os.listdir(feature_dir)] | |
| ) | |
| options = [] | |
| for name in names: | |
| if f0_method: | |
| options.append( | |
| f"{gt_wavs_dir}/{name}.wav|{feature_dir}/{name}.npy|{f0_dir}/{name}.wav.npy|{f0nsf_dir}/{name}.wav.npy|0" | |
| ) | |
| else: | |
| options.append(f"{gt_wavs_dir}/{name}.wav|{feature_dir}/{name}.npy|0") | |
| fea_dim = 256 if rvc_version == "v1" else 768 | |
| if f0_method: | |
| for _ in range(2): | |
| options.append( | |
| f"{current_directory}/logs/mute/0_gt_wavs/mute{sampling_rate}.wav|{current_directory}/logs/mute/3_feature{fea_dim}/mute.npy|{current_directory}/logs/mute/2a_f0/mute.wav.npy|{current_directory}/logs/mute/2b-f0nsf/mute.wav.npy|0" | |
| ) | |
| else: | |
| for _ in range(2): | |
| options.append( | |
| f"{current_directory}/logs/mute/0_gt_wavs/mute{sampling_rate}.wav|{current_directory}/logs/mute/3_feature{fea_dim}/mute.npy|0" | |
| ) | |
| shuffle(options) | |
| with open(f"{model_path}/filelist.txt", "w") as f: | |
| f.write("\n".join(options)) | |