Ph^ ^ d dl Zd dlmZmZ d dlZd dlZd dlm Z m
Z
d dl m Z G d de Z
y) N)CallableOptional) download_urlverify_str_arg)
VisionDatasetc
e Zd ZdZdZ ddedee dededee d df fd
Z
d ed e j fd Z
d efd
Zd efdZddZ xZS ) MovingMNISTa5 `MovingMNIST `_ Dataset.
Args:
root (string): Root directory of dataset where ``MovingMNIST/mnist_test_seq.npy`` exists.
split (string, optional): The dataset split, supports ``None`` (default), ``"train"`` and ``"test"``.
If ``split=None``, the full data is returned.
split_ratio (int, optional): The split ratio of number of frames. If ``split="train"``, the first split
frames ``data[:, :split_ratio]`` is returned. If ``split="test"``, the last split frames ``data[:, split_ratio:]``
is returned. If ``split=None``, this parameter is ignored and the all frames data is returned.
transform (callable, optional): A function/transform that takes in an torch Tensor
and returns a transformed version. E.g, ``transforms.RandomCrop``
download (bool, optional): If true, downloads the dataset from the internet and
puts it in root directory. If dataset is already downloaded, it is not
downloaded again.
zGhttp://www.cs.toronto.edu/~nitish/unsupervised_video/mnist_test_seq.npyNrootsplit split_ratiodownload transformreturnc t | || t j j | j
| j j | _ | j j d d | _ |
t |dd || _
t |t st dt! | d|cxk rdk sn t# d | d
|| _ |r| j' | j) s t+ d t- j. t1 j2 t j j | j | j }| j d k( r|d | j$ }n | j d
k( r|| j$ d }|j5 dd j7 d j9 | _ y )N)r /r )traintestz,`split_ratio` should be an integer, but got z:`split_ratio` should be `1 <= split_ratio <= 19`, but got z instead.z> %>>$'D r3 c , t | j S N)lenr0 r1 s r2 __len__zMovingMNIST.__len__N s 499~ r3 c t j j t j j | j | j
S r8 )r r existsr r r! r: s r2 r' zMovingMNIST._check_existsQ s. ww~~bggll4+<+rQ s% % C 5S
- S
r3