h< n S SK rS SKJr S SKJrJrJr S SKr S SK
r
S SK J r J
r
S SKJr " S S\5 rg) N)Path)CallableOptionalUnion) download_urlverify_str_arg)
VisionDatasetc
^ \ rS rSrSrSr SS\\\4 S\ \ S\
S\ S \ \ S
S4 U 4S jjjr
S \
S
\R 4S
jrS
\
4S jrS
\ 4S jrSS jrSrU =r$ ) MovingMNIST a `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.
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.
transform (callable, optional): A function/transform that takes in a torch Tensor
and returns a transformed version. E.g, ``transforms.RandomCrop``
zGhttp://www.cs.toronto.edu/~nitish/unsupervised_video/mnist_test_seq.npyNrootsplit split_ratiodownload transformreturnc > [ TU ] XS9 [ R R U R
U R R 5 U l U R R S5 S U l Ub
[ USS5 X l
[ U[ 5 ( d [ S[! U5 35 eSUs=:: a S:: d O [# S U S
35 eX0l U( a U R' 5 U R) 5 ( d [+ S 5 e[, R. " [0 R2 " [ R R U R U R 5 5 5 nU R S :X a US U R$ nO U R S
:X a X`R$ S nUR5 SS5 R7 S5 R9 5 U l g )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 r7 c , [ U R 5 $ N)lenr3 r4 s r5 __len__MovingMNIST.__len__O s 499~ r7 c [ R R [ R R U R U R
5 5 $ r= )r r existsr r" r$ r? s r5 r* MovingMNIST._check_existsR s. ww~~bggll4+<+r\ s+ , , C 5S
- S
r7