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"，則返回第一個分割影格 data[:, :split_ratio]。 如果 split="test"，則返回最後一個分割影格 data[:, split_ratio:]。 如果 split=None，則忽略此參數並返回所有影格資料。

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

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

特殊成員:

__getitem__(idx: int) → Tensor

參數:

idx (int) – 索引

返回值:

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

返回類型:

torch.Tensor