Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

replace pesq

This commit is contained in:
shahules786 2022-10-22 11:17:22 +05:30
parent 9b15534812
commit 05e40f84b6
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
enhancer/loss.py
View File

@ -3,7 +3,7 @@ import logging
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality from pesq import pesq
from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility
@ -122,7 +122,6 @@ class Pesq:
self.sr = sr self.sr = sr
self.name = "pesq" self.name = "pesq"
self.mode = mode self.mode = mode
self.pesq = PerceptualEvaluationSpeechQuality(fs=sr, mode=mode)
def __call__(self, prediction: torch.Tensor, target: torch.Tensor): def __call__(self, prediction: torch.Tensor, target: torch.Tensor):
@ -130,7 +129,12 @@ class Pesq:
for pred, target_ in zip(prediction, target): for pred, target_ in zip(prediction, target):
try: try:
pesq_values.append( pesq_values.append(
self.pesq(pred.squeeze(), target_.squeeze()).item() pesq(
self.sr,
target_.squeeze().detach().numpy(),
pred.squeeze().detach().numpy(),
self.mode,
)
) )
except Exception as e: except Exception as e:
logging.warning(f"{e} error occured while calculating PESQ") logging.warning(f"{e} error occured while calculating PESQ")

2
requirements.txt
View File

@ -5,7 +5,7 @@ joblib>=1.2.0
librosa>=0.9.2 librosa>=0.9.2
mlflow>=1.29.0 mlflow>=1.29.0
numpy>=1.23.3 numpy>=1.23.3
pesq==0.0.4 git+https://github.com/ludlows/python-pesq#egg=pesq
protobuf>=3.19.6 protobuf>=3.19.6
pystoi==0.3.3 pystoi==0.3.3
pytest-lazy-fixture>=0.6.3 pytest-lazy-fixture>=0.6.3