移动MNIST

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

MovingMNIST 数据集。

Parameters:

• root (str 或 pathlib.Path) – 数据集的根目录，其中存在 MovingMNIST/mnist_test_seq.npy。

• split (string, optional) – 数据集的分割，支持 None（默认）、"train" 和 "test"。 如果 split=None，则返回完整的数据。

• split_ratio (int, 可选) – 帧数的分割比例。如果 split="train"，则返回前分割帧 data[:, :split_ratio]。如果 split="test"，则返回后分割帧 data[:, split_ratio:]。如果 split=None，则忽略此参数并返回所有帧数据。

• transform (callable, optional) – 一个函数/变换，它接收一个 torch Tensor 并返回一个变换后的版本。例如，transforms.RandomCrop

• 下载 (bool, 可选) – 如果为真，则从互联网下载数据集并将其放入根目录。如果数据集已经下载，则不会再次下载。

Special-members:

__getitem__(idx: int) → Tensor

Parameters:

idx (int) – 索引

Returns:

视频帧（torch Tensor[T, C, H, W]）。T 是帧数。

Return type:

torch.Tensor