MNIST加载已训练好的模型进行预测
初学时练习用的代码,已加注释
基于tensorflow2
import tensorflow as tf
import numpy as np
import matplotlib.pylab as plt
from tensorflow.keras import datasets, layers, optimizers, Sequential, metrics
import scipy
import imageio
import cv2
from PIL import Image
from scipy import ndimage
import scipy.misc
(x, y), (x_val, y_val) = datasets.mnist.load_data()
for i in range(20):
image_array = x_val[i]
filename = "./images/" + 'mnist_test_%d.jpg' % i
img = Image.fromarray(np.uint8(image_array))
img.save(filename)