|
|
import os |
|
|
import shutil |
|
|
from random import shuffle |
|
|
from rvc.configs.config import Config |
|
|
import json |
|
|
|
|
|
config = Config() |
|
|
current_directory = os.getcwd() |
|
|
|
|
|
|
|
|
def generate_config(sample_rate: int, model_path: str): |
|
|
config_path = os.path.join("rvc", "configs", f"{sample_rate}.json") |
|
|
config_save_path = os.path.join(model_path, "config.json") |
|
|
if not os.path.exists(config_save_path): |
|
|
shutil.copyfile(config_path, config_save_path) |
|
|
|
|
|
|
|
|
def generate_filelist(model_path: str, sample_rate: int, include_mutes: int = 2): |
|
|
gt_wavs_dir = os.path.join(model_path, "sliced_audios") |
|
|
feature_dir = os.path.join(model_path, f"extracted") |
|
|
|
|
|
f0_dir, f0nsf_dir = None, None |
|
|
f0_dir = os.path.join(model_path, "f0") |
|
|
f0nsf_dir = os.path.join(model_path, "f0_voiced") |
|
|
|
|
|
gt_wavs_files = set(name.split(".")[0] for name in os.listdir(gt_wavs_dir)) |
|
|
feature_files = set(name.split(".")[0] for name in os.listdir(feature_dir)) |
|
|
|
|
|
f0_files = set(name.split(".")[0] for name in os.listdir(f0_dir)) |
|
|
f0nsf_files = set(name.split(".")[0] for name in os.listdir(f0nsf_dir)) |
|
|
names = gt_wavs_files & feature_files & f0_files & f0nsf_files |
|
|
|
|
|
options = [] |
|
|
mute_base_path = os.path.join(current_directory, "logs", "mute") |
|
|
sids = [] |
|
|
for name in names: |
|
|
sid = name.split("_")[0] |
|
|
if sid not in sids: |
|
|
sids.append(sid) |
|
|
options.append( |
|
|
f"{os.path.join(gt_wavs_dir, name)}.wav|{os.path.join(feature_dir, name)}.npy|{os.path.join(f0_dir, name)}.wav.npy|{os.path.join(f0nsf_dir, name)}.wav.npy|{sid}" |
|
|
) |
|
|
|
|
|
if include_mutes > 0: |
|
|
mute_audio_path = os.path.join( |
|
|
mute_base_path, "sliced_audios", f"mute{sample_rate}.wav" |
|
|
) |
|
|
mute_feature_path = os.path.join(mute_base_path, f"extracted", "mute.npy") |
|
|
mute_f0_path = os.path.join(mute_base_path, "f0", "mute.wav.npy") |
|
|
mute_f0nsf_path = os.path.join(mute_base_path, "f0_voiced", "mute.wav.npy") |
|
|
|
|
|
|
|
|
for sid in sids * include_mutes: |
|
|
options.append( |
|
|
f"{mute_audio_path}|{mute_feature_path}|{mute_f0_path}|{mute_f0nsf_path}|{sid}" |
|
|
) |
|
|
|
|
|
file_path = os.path.join(model_path, "model_info.json") |
|
|
if os.path.exists(file_path): |
|
|
with open(file_path, "r") as f: |
|
|
data = json.load(f) |
|
|
else: |
|
|
data = {} |
|
|
data.update( |
|
|
{ |
|
|
"speakers_id": len(sids), |
|
|
} |
|
|
) |
|
|
with open(file_path, "w") as f: |
|
|
json.dump(data, f, indent=4) |
|
|
|
|
|
shuffle(options) |
|
|
|
|
|
with open(os.path.join(model_path, "filelist.txt"), "w") as f: |
|
|
f.write("\n".join(options)) |
|
|
|
