| # pylint: skip-file |
| import os, gzip |
| import pickle as pickle |
| import sys |
| |
| # download mnist.pkl.gz |
| def GetMNIST_pkl(): |
| if not os.path.isdir("data/"): |
| os.system("mkdir data/") |
| if not os.path.exists('data/mnist.pkl.gz'): |
| os.system("wget -q http://deeplearning.net/data/mnist/mnist.pkl.gz -P data/") |
| |
| # download ubyte version of mnist and untar |
| def GetMNIST_ubyte(): |
| if not os.path.isdir("data/"): |
| os.system("mkdir data/") |
| if (not os.path.exists('data/train-images-idx3-ubyte')) or \ |
| (not os.path.exists('data/train-labels-idx1-ubyte')) or \ |
| (not os.path.exists('data/t10k-images-idx3-ubyte')) or \ |
| (not os.path.exists('data/t10k-labels-idx1-ubyte')): |
| os.system("wget -q http://data.mxnet.io/mxnet/data/mnist.zip -P data/") |
| os.chdir("./data") |
| os.system("unzip -u mnist.zip") |
| os.chdir("..") |
| |
| # download cifar |
| def GetCifar10(): |
| if not os.path.isdir("data/"): |
| os.system("mkdir data/") |
| if (not os.path.exists('data/cifar/train.rec')) or \ |
| (not os.path.exists('data/cifar/test.rec')) or \ |
| (not os.path.exists('data/cifar/train.lst')) or \ |
| (not os.path.exists('data/cifar/test.lst')): |
| os.system("wget -q http://data.mxnet.io/mxnet/data/cifar10.zip -P data/") |
| os.chdir("./data") |
| os.system("unzip -u cifar10.zip") |
| os.chdir("..") |
