GNSDR,GSIR,GSAR第一次

import librosa
from mir_eval.separation import bss_eval_sources
import numpy as np

sample_rate = 16000
mixed_wav1 = 'mix.wav'
pred_src1_wav1 = 'voice.wav'
pred_src2_wav1 = 'music.wav'
src1_wav1 = 'source-02.wav'
src2_wav1 = 'source-01.wav'

mixed_wav,_ = librosa.load(mixed_wav1,sr= sample_rate,mono=True)
pred_src1_wav,_ = librosa.load(pred_src1_wav1,sr= sample_rate,mono=True)
pred_src2_wav,_ = librosa.load(pred_src2_wav1,sr= sample_rate,mono=True)

src1_wav,_ = librosa.load(src1_wav1,sr= sample_rate,mono=True)

src2_wav,_ = librosa.load(src2_wav1,sr= sample_rate,mono=True)


def bss_eval_global(mixed_wav, src1_wav, src2_wav, pred_src1_wav, pred_src2_wav):
    len_cropped = pred_src1_wav.shape[-1]
    src1_wav = src1_wav[:len_cropped]
    src2_wav = src2_wav[:len_cropped]
    mixed_wav = mixed_wav[:len_cropped]
    gnsdr, gsir, gsar = np.zeros(2), np.zeros(2), np.zeros(2)
    total_len = 0
    # for i in range(2):
    sdr, sir, sar, _ = bss_eval_sources(np.array([src1_wav, src2_wav]),
                                        np.array([pred_src1_wav, pred_src2_wav]), True)
    sdr_mixed, _, _, _ = bss_eval_sources(np.array([src1_wav, src2_wav]),
                                          np.array([mixed_wav, mixed_wav]), True)
    nsdr = sdr - sdr_mixed
    gnsdr += len_cropped * nsdr
    gsir += len_cropped * sir
    gsar += len_cropped * sar
    total_len += len_cropped
    gnsdr = gnsdr / total_len
    gsir = gsir / total_len
    gsar = gsar / total_len
    print('GNSDR:', gnsdr)
    print('GSIR:', gsir)
    print('GSAR:', gsar)


if __name__ == '__main__':
    bss_eval_global(mixed_wav, src1_wav, src2_wav, pred_src1_wav, pred_src2_wav)

GNSDR: [245.99042613 253.92618371]
GSIR: [261.35599    243.66513372]
GSAR: [251.93602198 242.72301852]