如何在张量流中将图像张量转换为numpy数组？

Question

我正在训练CNN以生成图像。所有图像的类型均为张量。我希望将它们转换为numpy数组，然后可以使用opencv处理它们。

我知道.numpy()方法，它将我的张量转换为一个numpy数组，但是形状仍然是张量。我无法在cv2中使用它。

这是我的代码：

p=model_(x) 
s=p.numpy()
print(s.shape)
cv2.imwrite("hello.jpg",s)

（1，183，275，3），这是使用.numpy()生成的数组的形状，如何更改其形状以保留输出图像？

Answer 1

您需要摆脱第一个模糊（批处理），只需对reshape使用切片即可。

s=p.numpy()
print(s.shape)
cv2.imwrite("hello.jpg",s.reshape(s.shape[1:]))