MovingMNIST

class torchvision.datasets.MovingMNIST(root: Union[str, Path], split: Optional[str] = None, split_ratio: int = 10, download: bool = False, transform: Optional[Callable] = None)

MovingMNIST 資料集。

引數:

• root (str 或 pathlib.Path) – 資料集根目錄，其中包含 MovingMNIST/mnist_test_seq.npy。

• split (string, optional) – 資料集分割，支援 None（預設）、"train" 和 "test"。如果 split=None，則返回完整資料。

• split_ratio (int, optional) – 幀數的分割比例。如果 split="train"，則返回前 split 幀 data[:, :split_ratio]。如果 split="test"，則返回後 split 幀 data[:, split_ratio:]。如果 split=None，則忽略此引數並返回所有幀資料。

• download (bool, optional) – 如果為 True，則從網際網路下載資料集並將其放入根目錄。如果資料集已下載，則不會再次下載。

• transform (callable, optional) – 一個函式/轉換，接收一個 torch Tensor 並返回一個轉換後的版本。例如，transforms.RandomCrop

特殊成員:

__getitem__(idx: int) → Tensor

引數:

idx (int) – 索引

返回:

影片幀（torch Tensor[T, C, H, W]）。T 是幀數。

返回型別:

torch.Tensor