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

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-11-07 12:02:42 +05:30
parent ce04720e59 234e1a89de
commit 2d0b309b4d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/loss.py
View File

@ -3,6 +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 import ScaleInvariantSignalNoiseRatio
from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality
from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility
@ -188,4 +189,5 @@ LOSS_MAP = {
"si-sdr": Si_SDR, "si-sdr": Si_SDR,
"pesq": Pesq, "pesq": Pesq,
"stoi": Stoi, "stoi": Stoi,
"si-snr": ScaleInvariantSignalNoiseRatio,
} }