将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手写数据集转换成图片保存到本地的主要内容,如果未能解决你的问题,请参考以下文章
基于 Mindspore 框架与 ModelArts 平台的 MNIST 手写体识别实验