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 or Path) – 数据集顶层目录所在的根目录

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

• sources (List[str] or None, optional) – 用于提取数据的源。列表可包含以下选项：["bass", "drums", "other", "mixture", "vocals"]。如果为 None，则数据集包含除混合音轨之外的所有音轨。（默认：None）

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

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

__getitem__

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

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

参数

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

返回

元组包含以下项：

张量

波形

int

采样率

int

帧数

str

曲目名称