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, 可選) – 資料集分割，支援 None (預設)、"train" 和 "test"。如果 split=None，則返回全部資料。

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

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

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

特殊成員：

__getitem__(idx: int) → Tensor

引數：

idx (int) – 索引

返回：

影片幀 (torch Tensor[T, C, H, W])。其中 T 是幀數。

返回型別：

torch.Tensor