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 (字符串, 可选) – 数据集分割，支持 None（默认）、"train" 和 "test"。如果 split=None，则返回完整数据。

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

• transform (可调用对象, 可选) – 一个函数/转换，它接收一个 torch 张量并返回一个转换后的版本。例如，transforms.RandomCrop

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

特殊成员：

__getitem__(idx: int) → Tensor

参数：

idx (整数) – 索引

返回值：

视频帧 (torch 张量[T, C, H, W])。其中 T 是帧数。

返回类型：

torch.Tensor