machine-learning - 如何将张量写入 PNG 图像文件？

Question

我正在尝试将具有以下属性的张量写入 PNG 文件。

type(adv_x) 
<class 'tensorflow.python.framework.ops.EagerTensor'>

adv_x.shape
(1, 600, 800, 3)

tf.rank(adv_x) 
tf.Tensor(4, shape=(), dtype=int32)

我知道如何将张量显示为图像 (plt.imshow(adv_x[0]))，但我想将其写入文件，以便最终得到 RGB .PNG文件大小为 600 x 800 像素。

提前致谢。

Answer 1

留在 TF 领域通常是更明智的选择。因此，在 TF 2.0 中您也可以简单地使用

tf.keras.preprocessing.image.save_img('path/to/file/file.png',adv_x[0])

它在后台使用 PIL 并为您提供所有灵 active ，但不需要额外导入或转换张量。