将MNIST手写数据集转换成图片保存到本地

Posted xxmmqg

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了将MNIST手写数据集转换成图片保存到本地相关的知识,希望对你有一定的参考价值。

# 加载图片
data = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = data.load_data()
plt.imshow(x_train[0], cmap=‘gray‘)
技术图片
# 创建10个文件夹存放每一类图片
for i in range(10):
    os.makedirs(f"../datas/mnist/train/{i}")
    os.makedirs(f"../datas/mnist/test/{i}")
# 保存图片并生成图像列表

# 训练集数据
with open("../datas/mnist/train/image_list.txt", ‘w‘) as img_list:
    i = 1
    for img, label in zip(x_train, y_train):
        img = Image.fromarray(img) # 将array转化成图片
        img_save_path = f"../datas/mnist/train/{label}/{i}.jpg" # 图片保存路径
        img.save(img_save_path) # 保存图片
        img_list.write(img_save_path + "	" + str(label) + "
")
        i += 1

# 测试集数据
with open("../datas/mnist/test/image_list.txt", ‘w‘) as img_list:
    i = 1
    for img, label in zip(x_test, y_test):
        img = Image.fromarray(img) # 将array转化成图片
        img_save_path = f"../datas/mnist/test/{label}/{i}.jpg" # 图片保存路径
        img.save(img_save_path)
        img_list.write(img_save_path + "	" + str(label) + "
")
        i += 1

以上是关于将MNIST手写数据集转换成图片保存到本地的主要内容,如果未能解决你的问题,请参考以下文章

基于MNIST数据集实现手写数字识别

神经网络做MNIST手写数字识别代码

基于 Mindspore 框架与 ModelArts 平台的 MNIST 手写体识别实验

在PaddlePaddle上实现MNIST手写体数字识别

Keras训练神经网络DEMO——全连接神经网络训练MNIST

Keras训练神经网络DEMO——全连接神经网络训练MNIST