在Python中如何保存图像,如何将数组保存为图像
主要是看了TensorFlow中第一部分中看到的,查了查资料总结下如何将数组保存为图像
原程序如下
import scipy.misc
import os
#from PIL import Image
#把原始图片保存至MNIST_data/raw/下
save_dir="MNIST_data/raw/"
if os.path.exists(save_dir)is False:
os.makedirs(save_dir)
for i in range(20):
image_array=mnist.train.images[i,:]#打印第i个
image_array=image_array.reshape(28,28)
filename=save_dir + 'mnist_train_%d.jpeg' % i#保存文件的格式
print(filename)
print(mnist.train.labels[i,:])#打印出标签
scipy.misc.toimage(image_array,cmin=0.0,cmax=1.0).save(filename)#保存图像
程序运行结果如下:
MNIST_data/raw/mnist_train_0.jpeg
[0. 0. 0. 0. 0. 0. 0. 1. 0. 0.]
MNIST_data/raw/mnist_train_1.jpeg
[0. 0. 0. 1. 0. 0. 0. 0. 0. 0.]
MNIST_data/raw/mnist_train_2.jpeg
[0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]
MNIST_data/raw/mnist_train_3.jpeg
[0. 0. 0. 0. 0. 0. 1. 0. 0. 0.]
MNIST_data/raw/mnist_tra