SquimObjective

class torchaudio.models.SquimObjective(encoder: Module, dprnn: Module, branches: ModuleList)

Speech Quality and Intelligibility Measures (SQUIM) 模型，用于预测语音增强的客观指标得分（例如，STOI、PESQ 和 SI-SDR）。

参数

• encoder (torch.nn.Module) – 将 1D 波形转换为 2D 特征表示的编码器模块。

• dprnn (torch.nn.Module) – 用于建模序列特征的 DPRNN 模块。

• branches (torch.nn.ModuleList) – Transformer 分支，每个分支估计一个客观指标得分。

SquimObjective 使用的教程

Torchaudio-Squim：TorchAudio 中的非侵入式语音评估

Torchaudio-Squim：TorchAudio 中的非侵入式语音评估

方法

forward

SquimObjective.forward(x: Tensor) → List[Tensor]

参数

x (torch.Tensor) – 输入波形。维度为 (batch, time) 的张量。

返回

得分张量列表。每个张量的维度为 (batch,)。

返回类型

List(torch.Tensor)

工厂函数

squim_objective_model	构建自定义 torchaudio.prototype.models.SquimObjective 模型。
squim_objective_base	使用默认参数构建 torchaudio.prototype.models.SquimObjective 模型。