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_ratio 帧 data[:, :split_ratio]。如果 split="test"，则返回最后 split_ratio 帧 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