From 05e40f84b6eefee8831508e90252cd8f9cca7bc2 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 22 Oct 2022 11:17:22 +0530 Subject: [PATCH] replace pesq --- enhancer/loss.py | 10 +++++++--- requirements.txt | 2 +- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/enhancer/loss.py b/enhancer/loss.py index 5092656..f2be8df 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -3,7 +3,7 @@ import logging import numpy as np import torch import torch.nn as nn -from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality +from pesq import pesq from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility @@ -122,7 +122,6 @@ class Pesq: self.sr = sr self.name = "pesq" self.mode = mode - self.pesq = PerceptualEvaluationSpeechQuality(fs=sr, mode=mode) def __call__(self, prediction: torch.Tensor, target: torch.Tensor): @@ -130,7 +129,12 @@ class Pesq: for pred, target_ in zip(prediction, target): try: 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: logging.warning(f"{e} error occured while calculating PESQ") diff --git a/requirements.txt b/requirements.txt index 95f145d..fa5e41c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ joblib>=1.2.0 librosa>=0.9.2 mlflow>=1.29.0 numpy>=1.23.3 -pesq==0.0.4 +git+https://github.com/ludlows/python-pesq#egg=pesq protobuf>=3.19.6 pystoi==0.3.3 pytest-lazy-fixture>=0.6.3