How to load an image and show the image using keras?

user_1159517 0
Viewed 31800
%matplotlib inline
from keras.preprocessing import image

import matplotlib.pyplot as plt
import numpy as np
img = np.random.rand(224,224,3)
plt.imshow(img)
plt.show()

img_path = "image.jpeg"
img = image.load_img(img_path, target_size=(224, 224))
print(type(img))

x = image.img_to_array(img)
print(type(x))
print(x.shape)
plt.imshow(x)

I have some code like this which should print the image. But it shows the image in wrong channels. What am i missing here?

keras tensorflow matplotlib numpy python
2 Answers

This question is kind of old, but there is a very comfortable way to display images:

tf.keras.preprocessing.image.array_to_img(image[0]).show()

Your image has to have 3 dimensions (if its in a batch as normally, just take desired_element). Works fine on EagerTensors or numpy arrays.

user_96210800
Related