How to convert image tensor in to numpy array in tensorflow?

Question

I am training a CNN to generate images. The type of all the images are tensors. I want them to be converted into numpy arrays then I can process them using opencv.

I know about the .numpy() method, it converts my tensor into an numpy array but the shape is still tensor. I can't get it to work in cv2.

Here is my code:

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

(1, 183, 275, 3), this is the shape of the array generated using .numpy(), how can I change its shape to retain output image?

Zabir Al Nazi Nabil · Accepted Answer · 2020-09-02 12:35:03Z

3

You need to get rid of the first dim (batch), just use slicing with reshape.

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

answered Sep 2, 2020 at 12:35

Zabir Al Nazi Nabil

11.3k4 gold badges38 silver badges71 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to convert image tensor in to numpy array in tensorflow?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related