eval/TTS_speech_quality.py

from utils import int2float
def TTS_psuedomos(TTS_audio_output):
    from versa import pseudo_mos_metric, pseudo_mos_setup, sheet_ssqa, sheet_ssqa_setup

    predictor_dict, predictor_fs = pseudo_mos_setup(
        use_gpu=True,
        predictor_types=["utmos", "dnsmos", "plcmos"],
        predictor_args={"utmos":{"fs": 16000},"dnsmos":{"fs": 16000},"plcmos":{"fs": 16000}},
    )
    score_modules = {
        "module": pseudo_mos_metric,
        "args": {
            "predictor_dict": predictor_dict,
            "predictor_fs": predictor_fs,
            "use_gpu": True,
        },
    }
    dict1=score_modules["module"](
            int2float(TTS_audio_output[1]),
            TTS_audio_output[0],
            **score_modules["args"],
    )
    str1=""
    for k in dict1:
        str1=str1+f"{k}: {dict1[k]:.2f}\n"
    sheet_model = sheet_ssqa_setup(
        model_tag="default",
        model_path=None,
        model_config=None,
        use_gpu=True,
    )
    score_modules = {
        "module": sheet_ssqa,
        "args": {"model": sheet_model, "use_gpu": True},
    }
    dict1 = score_modules["module"](
        score_modules["args"]["model"], int2float(TTS_audio_output[1]), TTS_audio_output[0],
        use_gpu=score_modules["args"]["use_gpu"]
    )
    for k in dict1:
        str1=str1+f"{k}: {dict1[k]:.2f}\n"
    return str1