MUSDB_HQ

class torchaudio.datasets.MUSDB_HQ(root: Union[str, Path], subset: str, sources: Optional[List[str]] = None, split: Optional[str] = None, download: bool = False)

MUSDB_HQ [Rafii et al., 2019] 数据集。

参数:

• root (str 或 Path) – 找到数据集顶层目录的根目录

• subset (str) – 要使用的数据集子集。选项：["train", "test"]。

• sources (List[str] 或 None, 可选) – 从中提取数据的源。列表可以包含以下选项：["bass", "drums", "other", "mixture", "vocals"]。如果为 None，则数据集由音轨组成，混合音轨除外。（默认值：None）

• split (str 或 None, 可选) – 是否将训练集拆分为训练集和验证集。如果为 None，则不进行拆分。如果为 train 或 validation，则返回相应的集合。（默认值：None）

• download (bool, 可选) – 如果在根路径中未找到数据集，是否下载数据集。（默认值：False）

__getitem__

MUSDB_HQ.__getitem__(n: int) → Tuple[Tensor, int, int, str]

从数据集中加载第 n 个样本。

参数:

n (int) – 要加载的样本的索引

返回:

以下项目的元组；

Tensor

波形

int

采样率

int

帧数

str

音轨名称