j v d dl Zd dlmZ d dlmZmZmZ d dlZ d dl
Z
d dl m Z m
Z
d dlmZ G d de ZdS ) N)Path)CallableOptionalUnion) download_urlverify_str_arg)
VisionDatasetc
e Zd ZdZdZ ddeeef dee de d e
d
ee d df fd
Z d
e d e
j fdZd e fdZd e
fdZddZ xZS ) MovingMNISTaE `MovingMNIST `_ Dataset.
Args:
root (str or ``pathlib.Path``): 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 a 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.npyN
Frootsplit split_ratiodownload transformreturnc t || t j | j | j j | _ | j
d d | _ |t |dd || _
t |t s t dt! | d|cxk rdk sn t# d | d
|| _ |r| | st+ d t- j t1 j t j | j | j }| j
d k r|d | j }n| j
d
k r|| j d }| dd d | _ d S )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 r6 c * t | j S N)lenr3 r4 s r5 __len__zMovingMNIST.__len__O s 49~~ r6 c t j t j | j | j S r; )r r existsr r" r$ r= s r5 r* zMovingMNIST._check_existsR s+ w~~bgll4+ r* r
__classcell__)r s @r5 r r s/ UD
$ (,
"C "CCI "C }"C "C
"C H%
"C
"C "C "C "C "C "CH s u| Ot O O O O
r6 r )os.pathr pathlibr typingr r r numpyr. r, torchvision.datasets.utilsr r torchvision.datasets.visionr r r6 r5 rT s , , , , , , , , , , C C C C C C C C 5 5 5 5 5 5S
S
S
S
S
- S
S
S
S
S
r6