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 等人，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

曲目名称