mlp/Mnist.py at master · koolok/mlp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Nov 14 14:44:39 2017

@author: koolok
"""

import mnist
from PIL import Image
import pickle


def mnist_to_pickle() :
    train_images = mnist.train_images()

    images = []

    for i in range(len(train_images)) :
        data = train_images[i,:,:] * -1 + 256

        dataL = []
        dataL.extend([256]*30)

        for line in data :
            line = list(line)
            line.insert(0,256)
            line.append(256)
            dataL.extend(line)

        dataL.extend([256]*30)

        imageL = Image.new("L",(30,30))
        imageL.putdata(dataL)

        imageRGB = imageL.convert(mode="RGB")


        images.append(imageRGB)


    file = open('mnist_train.pk', 'wb')

    pickle.dump(images, file)
    file.close()


    test_images = mnist.test_images()

    images = []

    for i in range(len(test_images)) :
        data = test_images[i,:,:] * -1 + 256

        dataL = []
        dataL.extend([256]*30)

        for line in data :
            line = list(line)
            line.insert(0,256)
            line.append(256)
            dataL.extend(line)

        dataL.extend([256]*30)

        imageL = Image.new("L",(30,30))
        imageL.putdata(dataL)

        imageRGB = imageL.convert(mode="RGB")

        images.append(imageRGB)


    file = open('mnist_test.pk', 'wb')

    pickle.dump(images, file)
    file.close()

def load_mnist() :
    try :
        file = open("mnist_train.pk", 'rb')
        train_images = pickle.load(file)
        file.close()

        file = open("mnist_test.pk", 'rb')
        test_images = pickle.load(file)
        file.close()

        return train_images, mnist.train_labels(), test_images, mnist.test_labels()

    except :
        print("mnist_train.pk or mnist_test.pk not found. Please run mnist_to_pickle")