python把keras的mnist数据集保存为jpg图片格式

python把keras的mnist数据集保存为jpg图片格式

最近需要对mnist图片进行处理，但是mnist数据集似乎都已经封装好了，我分享一下我怎么解析生jpg图片的， 我的环境是：

Keras 2.3.0Keras-Applications 1.0.8Keras-Preprocessing 1.1.2

下面是代码：

import kerasfrom keras.datasets import mnistimport numpy as npfrom PIL import Image, ImageOpsimport osdef save_image(filename, data_array): im = Image.fromarray(data_array.astype('uint8')) im_invert = ImageOps.invert(im) im_invert.save(filename)# Load MNIST Data(x_train, y_train), (x_test, y_test) = mnist.load_data()DIR_NAME = "mnist_images"if os.path.exists(DIR_NAME) == False: os.mkdir(DIR_NAME)# Save Imagesi = 0for data_x,data_y in [[x_train,y_train], [x_test,y_test]]: print("[---------------------------------------------------------------]") for x,y in zip(data_x,data_y): file_path=os.path.join(DIR_NAME,str(y)) os.makedirs(file_path,exist_ok=True) filename = "{0}/{1:05d}.jpg".format(file_path,i) print(filename) save_image(filename, x) i += 1

然后就会在本地生成mnist图片格式的数据集了哈。

参考文献

​​mnist_to_jpg.py​​

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。