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

音軌名稱