目录
1.显示MNIST图像
import numpy as np
from dataset.mnist import load_mnist
from PIL import Image
def img_show(img):
pil_img = Image.fromarray(np.uint8(img))
pil.img.show()
(x_train, t_train), (x_test, t_test) = load_mnist(flatten=True, normalize=False)
img = x_train[0]
label = t_train[0]
print(label)
img = img.reshape(28, 28)
img_show(img)
2.对MNIST数据集实现神经网络的推理处理
import numpy as np
import pickle
from dataset.mnist import load_mnist
def sigmoid(x):
return 1 / (1 + np.exp(-x))
def softmax(x):
a = np.max(x)
exp_a = np.exp(x - a)
sum_exp_a = np.sum(exp_a)
y = exp_a / sum_exp_a
return y
def get_data():
(x_train, t_train), (x_test, t_test) = load_mnist(normalize=True, flatten=True, one_hot_label=False)
return x_test, t_test
def init_network():
with open('sample_weight.pkl', 'rb') as f:
network = pickle.load(f)
def predict(network, x):
W1, W2, W3 = network['W1'], network['W2'], network['W3']
b1, b2, b3 = network['b1'], network['b2'], network['b3']
a1 = np.dot(x, W1) + b1
z1 = sigmoid(a1)
a2 = np.dot(z1, W2) + b2
z2 = sigmoid(a2)
a3 = np.dot(z2, W3) + b3
y = softmax(a3)
return y
x, t = get_data()
network = init_network()
accuracy_cnt = 0
for i in range(len(x)):
y = predict(network, x)
p = np.argmax(y)
if p == t[i]:
accuracy_cnt += 1
print('Accuracy:' + str(float(accuracy_int) / len(x))
通过批处理方法,我们可以把后面部分代码更改如下:
x, t = get_data()
network = init_network()
batch_size = 100
accuracy_int = 0
for i in range(0, len(x), batch_size):
x_batch = x[i, i+batchsize]
y_batch = predict(network, x_batch)
p = np.argmax(y_batch, axis=1)
accuracy_cnt += np.sum(p == t[i:i+batch_size]
print('Accuracy:' + str(float(accuracy_cnt) / len(x)))