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

音轨名称