speechbrain.inference.metrics 模块

指定用于度量估计模块的推理接口。

作者

  • Aku Rouhe 2021

  • Peter Plantinga 2021

  • Loren Lugosch 2020

  • Mirco Ravanelli 2020

  • Titouan Parcollet 2021

  • Abdel Heba 2021

  • Andreas Nautsch 2022, 2023

  • Pooneh Mousavi 2023

  • Sylvain de Langen 2023

  • Adel Moumen 2023

  • Pradnya Kandarkar 2023

摘要

类

SNREstimator

一个“即用型”信噪比 (SNR) 估计器。

参考

class speechbrain.inference.metrics.SNREstimator(modules=None, hparams=None, run_opts=None, freeze_params=True)[源码]

基类：Pretrained

一个“即用型”信噪比 (SNR) 估计器。

MODULES_NEEDED = ['encoder', 'encoder_out']
HPARAMS_NEEDED = ['stat_pooling', 'snrmax', 'snrmin']
estimate_batch(mix, predictions)[源码]

对估计的源和混合信号运行 SI-SNR 估计。

参数：

  • mix (torch.Tensor) – 形状为 B X T 的源混合信号

  • predictions (torch.Tensor) –

    大小为 (B x T x C)，其中 B 为批处理大小

    T 为时间点数，C 为源数量

返回：

SNR 的估计值

返回类型：

tensor

forward(mix, predictions)[源码]

只需运行批处理估计

gettrue_snrrange(inp)[源码]

从 0-1 范围转换为真实的 snr 范围