Spaces:
Running
on
T4
Running
on
T4
File size: 700 Bytes
32ca76b |
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 |
import pypesq
import numpy as np
def batch_pesq(batch_signal, batch_signal_wmd):
batch_signal1 = batch_signal.detach().cpu().numpy()
batch_signal2 = batch_signal_wmd.detach().cpu().numpy()
pesq_array = []
for signal1, signal2 in zip(batch_signal1, batch_signal2):
try:
pesq = pypesq.pesq(signal1, signal2, 16000)
#可能会有错误：ValueError: ref is all zeros, processing error!
except Exception as e:
print(e)
continue
if np.isnan(pesq):
print("pesq is nan!")
continue
pesq_array.append(pesq)
if len(pesq_array) > 0:
return np.mean(pesq_array)
return -1
|