speechbrain.inference.metrics 模块

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

Authors:

  • 阿库·罗赫 2021

  • 彼得·普兰廷加 2021

  • 洛伦·卢戈斯奇 2020

  • Mirco Ravanelli 2020

  • Titouan Parcollet 2021

  • 阿卜杜勒·赫巴 2021

  • 安德烈亚斯·诺茨 2022, 2023

  • Pooneh Mousavi 2023

  • Sylvain de Langen 2023

  • 阿德尔·穆门 2023

  • 普拉迪亚·坎达尔卡 2023

摘要

类：

SNREstimator

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

参考

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

基础类：Pretrained

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

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

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

Parameters:

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

  • predictions (torch.Tensor) –

    大小为 (B x T x C), 其中 B 是批量大小

    T 是时间点的数量 C 是源的数量

Returns:

信噪比估计

Return type:

张量

forward(mix, predictions)[source]

只需运行批量估计

gettrue_snrrange(inp)[source]

将0-1范围转换为真实的信噪比范围