我正在训练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()
生成的数组的形状,如何更改其形状以保留输出图像?
最佳答案
您需要摆脱第一个dim(批处理),只需对reshape
使用 slice 即可。
s=p.numpy()
print(s.shape)
cv2.imwrite("hello.jpg",s.reshape(s.shape[1:]))
关于python - 如何在 tensorflow 中将图像张量转换为numpy数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63705534/