Spaces:
Build error
Build error
File size: 4,097 Bytes
e882f51 |
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 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
import json
import os
import shutil
from functools import reduce
from pathlib import Path
import matplotlib
import matplotlib.pyplot as plt
import yaml
from pylab import xticks, np
from tqdm import tqdm
from modules.vocoders.nsf_hifigan import NsfHifiGAN
from preprocessing.process_pipeline import get_pitch_parselmouth, get_pitch_crepe
from utils.hparams import set_hparams, hparams
head_list = ["C", "C#", "D", "D#", "E", "F", "F#", "G", "G#", "A", "A#", "B"]
def compare_pitch(f0_static_dict, pitch_time_temp, trans_key=0):
return sum({k: v * f0_static_dict[str(k + trans_key)] for k, v in pitch_time_temp.items() if
str(k + trans_key) in f0_static_dict}.values())
def f0_to_pitch(ff):
f0_pitch = 69 + 12 * np.log2(ff / 440)
return round(f0_pitch, 0)
def pitch_to_name(pitch):
return f"{head_list[int(pitch % 12)]}{int(pitch / 12) - 1}"
def get_f0(audio_path, crepe=False):
wav, mel = NsfHifiGAN.wav2spec(audio_path)
if crepe:
f0, pitch_coarse = get_pitch_crepe(wav, mel, hparams)
else:
f0, pitch_coarse = get_pitch_parselmouth(wav, mel, hparams)
return f0
def merge_f0_dict(dict_list):
def sum_dict(a, b):
temp = dict()
for key in a.keys() | b.keys():
temp[key] = sum([d.get(key, 0) for d in (a, b)])
return temp
return reduce(sum_dict, dict_list)
def collect_f0(f0):
pitch_num = {}
pitch_list = [f0_to_pitch(x) for x in f0[f0 > 0]]
for key in pitch_list:
pitch_num[key] = pitch_num.get(key, 0) + 1
return pitch_num
def static_f0_time(f0):
if isinstance(f0, dict):
pitch_num = merge_f0_dict({k: collect_f0(v) for k, v in f0.items()}.values())
else:
pitch_num = collect_f0(f0)
static_pitch_time = {}
sort_key = sorted(pitch_num.keys())
for key in sort_key:
static_pitch_time[key] = round(pitch_num[key] * hparams['hop_size'] / hparams['audio_sample_rate'], 2)
return static_pitch_time
def get_end_file(dir_path, end):
file_lists = []
for root, dirs, files in os.walk(dir_path):
files = [f for f in files if f[0] != '.']
dirs[:] = [d for d in dirs if d[0] != '.']
for f_file in files:
if f_file.endswith(end):
file_lists.append(os.path.join(root, f_file).replace("\\", "/"))
return file_lists
if __name__ == "__main__":
# 给config文件增加f0_static统计音域
config_path = "F:/sovits/diff-svc-main/checkpoints/aquapre/config.yaml"
hparams = set_hparams(config=config_path, exp_name='', infer=True, reset=True, hparams_str='', print_hparams=False)
f0_dict = {}
# 获取batch文件夹下所有wav文件
wav_paths = get_end_file("F:/sovits/diff-svc-main/batch/aquapre", "wav")
# parselmouth获取f0
with tqdm(total=len(wav_paths)) as p_bar:
p_bar.set_description('Processing')
for wav_path in wav_paths:
f0_dict[wav_path] = get_f0(wav_path, crepe=False)
p_bar.update(1)
pitch_time = static_f0_time(f0_dict)
total_time = round(sum(pitch_time.values()), 2)
pitch_time["total_time"] = total_time
print(f"total time: {total_time}s")
shutil.copy(config_path, f"{Path(config_path).parent}\\back_{Path(config_path).name}")
with open(config_path, encoding='utf-8') as f:
_hparams = yaml.safe_load(f)
_hparams['f0_static'] = json.dumps(pitch_time)
with open(config_path, 'w', encoding='utf-8') as f:
yaml.safe_dump(_hparams, f)
print("原config文件已在原目录建立备份：back_config.yaml")
print("音域统计已保存至config文件，此模型可使用自动变调功能")
matplotlib.use('TkAgg')
plt.title("数据集音域统计", fontproperties='SimHei')
plt.xlabel("音高", fontproperties='SimHei')
plt.ylabel("时长(s)", fontproperties='SimHei')
xticks_labels = [pitch_to_name(i) for i in range(36, 96)]
xticks(np.linspace(36, 96, 60, endpoint=True), xticks_labels)
plt.plot(pitch_time.keys(), pitch_time.values(), color='dodgerblue')
plt.show()
|