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) – 要加载的样本的索引

返回

元组包含以下项：

张量

波形

int

采样率

int

帧数

str

音轨名称