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，则忽略此参数并返回所有帧数据。

• transform (callable, 可选) – 接收 torch Tensor 并返回转换后版本的函数/变换。例如, transforms.RandomCrop

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

特殊成员:

__getitem__(idx: int) → Tensor

参数:

idx (int) – 索引

返回值:

视频帧 (torch Tensor[T, C, H, W])。T 是帧数。

返回类型:

torch.Tensor