| 763 | train_image_data, train_label_data, test_image_data, test_label_data, image_data, label_data |
| 764 | """ |
| 765 | def __init__(self): |
| 766 | super(MNIST, self).__init__( |
| 767 | "mnist", |
| 768 | urls={ |
| 769 | "train_image_data": "http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz", |
| 770 | "train_label_data": "http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz", |
| 771 | "test_image_data": "http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz", |
| 772 | "test_label_data": "http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz", |
| 773 | "image_data": [], # depends on `train_image_data` & `test_image_data` |
| 774 | "label_data": [] # depends on `train_label_data` & `test_label_data` |
| 775 | } |
| 776 | ) |
| 777 | |
| 778 | def train_image_data_preprocess(self, raw_file, save_file): |
| 779 | images = np.fromfile(raw_file, dtype=np.uint8) |