Spaces:
Runtime error
Runtime error
File size: 2,709 Bytes
a8c39f5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
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(rvc_version: str, sample_rate: int, model_path: str):
config_path = os.path.join("rvc", "configs", rvc_version, 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, rvc_version: str, sample_rate: int):
gt_wavs_dir = os.path.join(model_path, "sliced_audios")
feature_dir = os.path.join(model_path, f"{rvc_version}_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"{gt_wavs_dir}/{name}.wav|{feature_dir}/{name}.npy|{f0_dir}/{name}.wav.npy|{f0nsf_dir}/{name}.wav.npy|{sid}"
)
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"{rvc_version}_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")
# always adding two files
for sid in sids:
options.append(
f"{mute_audio_path}|{mute_feature_path}|{mute_f0_path}|{mute_f0nsf_path}|{sid}"
)
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))
|