tensorflow2.0保存图片时老出错,查阅很多资料之后终于改好了:
首先看报错:
下面是正确的代码:
X = ( image_data[0] * 0.5 + 0.5 ) * 255 #归一化之后的图片数组,如果你的数组是正确的,就不需要归一化。
print(X)#输出x,观察自己的数组类型,我的是tf.Tensor类型,所以张量转换类型很重要
img = tf.cast(X, dtype=tf.uint8)# 这一步转换张量数据类型很重要
img = tf.image.encode_png(img)# 编码回图片
savepath = "E:\\..\\..\\"#写自己保存图片的地址
savepicture_name = savepath + str(ii + 1) + '.jpg' #字符串拼接保存图像的地址,ii指循环,如果要保存多张图片则可以加上for循环语句使用,保存一张就去掉ii. for 循环语句的位置要自己考虑一下。
with tf.io.gfile.GFile(savepicture_name, 'wb') as file:# 保存
file.write(img.numpy())
**注释:**输出的X的数组类型,是张量类型,所以保存时要转换:下图红色笔已经标出来
方法已经分享了,但具体还是要自己对照修改哟。希望可以帮助到你。