python - 如何将张量视为图像？

Question

我用tensorflow学习了一些数据。

对于测试，我看到了最终结果的形状。

它是 (1, 80, 80, 1) 的张量。

我使用 matplotlib 或 PIL 来执行此操作，

我想看看更改为饼图数组后的图像。

但我无法将张量更改为 numpy。

即使我使用了 eval()，由于 session 的原因我也无法做任何事情。

无法将张量转换为 numpy。

我可以将张量视为图像吗？

(mytensor1) # mytensor

arr = np.ndarray(mytensor1)
arr_ = np.squeeze(arr)
plt.imshow(arr_)
plt.show()

但有错误消息:类型错误:预期的序列对象 len >= 0 或单个整数

Answer 1

您可以使用 numpy 中的挤压函数。例如

arr = np.ndarray((1,80,80,1))#This is your tensor
arr_ = np.squeeze(arr) # you can give axis attribute if you wanna squeeze in specific dimension
plt.imshow(arr_)
plt.show()

现在，您可以轻松显示此图像(例如上面的代码，假设您使用 matplotlib.pyplot as plt)。