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

add si-snr

This commit is contained in:
shahules786 2022-11-07 12:01:20 +05:30
parent 7f3dcf39c5
commit 47cfc84295
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,
} }