如何在张量流中将图像张量转换为 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:]))