python中将手写数据数组转换为图片输入出来

最新推荐文章于 2024-07-05 02:22:31 发布

甄同学

最新推荐文章于 2024-07-05 02:22:31 发布

阅读量2.8k

点赞数

本文链接：https://blog.csdn.net/qq_25964837/article/details/79810328

版权

该博客介绍了如何使用Python的Keras库和相关模块将MNIST数据集的手写数字数组转换为图片，保存到本地文件系统。

摘要由CSDN通过智能技术生成

from __future__ import print_function
import keras
from keras.datasets import mnist
from keras.models import Model
from keras.layers import Input, Activation, add, Dense, Flatten, Dropout, Multiply, Embedding, Lambda
from keras.layers import Conv2D, MaxPooling2D,PReLU
from keras import backend as K
import numpy as np
import sys
import scipy.misc
from keras.optimizers import SGD, Adam


batch_size = 128
num_classes = 10
epochs = 50

#isCenterloss = True
isCenterloss = False



# input image dimensions
img_rows, img_cols = 28, 28

# the data, shuffled and split between train and test sets
(x_train, y_train), (x_test, y_test) = mnist.load_data()

if K.image_data_format() == 'channels_first':
    x_train = x_train.reshape(x_train.shape[0], 1, img_rows, img_cols)
    x_test = x_test.reshape(x_test.shape[0], 1, img_rows, img_cols)
    input_shape = (1, img_rows, img_cols)
else:
    x_train = x_trai