Constui o meu modelo basedo no curso, treinei e esta como o da professora. Agora como posso carregar a foto de uma saia por exemplo para que meu modelo possa classificar?
Constui o meu modelo basedo no curso, treinei e esta como o da professora. Agora como posso carregar a foto de uma saia por exemplo para que meu modelo possa classificar?
Olá Camila.
Primeiro vamos utilizar a função load_img do Keras, passando como parâmetro o nome da imagem, no modo de cor grayscale e tambem o tamanho [28,28] como foi explicado pela instrutora.
from keras.preprocessing.image import load_img
# carrega imagem
img = load_img('imagem.jpg', color_mode = "grayscale", target_size=[28, 28])
Depois vamos converter essa imagem para um array e fazer um reshape para [28, 28] novamente, já que o nosso img_to_array gerou uma dimensão a mais.
from keras.preprocessing.image import load_img
from keras.preprocessing.image import img_to_array
# carrega imagem
img = load_img('imagem.jpg', color_mode = "grayscale", target_size=[28, 28])
# convert imagem para um array numpy
img_array = img_to_array(img).astype(np.uint8)
img_array = img_array.reshape((28, 28))
Também inclui a criação de um array numpy que vai conter a sua imagem, você pode colocar mais imagens se quiser, fiz isso porque o predict espera um array numpy
from keras.preprocessing.image import load_img
from keras.preprocessing.image import img_to_array
# carrega imagem
img = load_img('imagem.jpg', color_mode = "grayscale", target_size=[28, 28])
# convert imagem para um array numpy
img_array = img_to_array(img).astype(np.uint8)
img_array = img_array.reshape((28, 28))
#conversão para um array numpy
lista = np.array([img_array])
Finalmente podemos prever com o nosso modelo usando a imagem.
from keras.preprocessing.image import load_img
from keras.preprocessing.image import img_to_array
# carrega imagem
img = load_img('imagem.jpg', color_mode = "grayscale", target_size=[28, 28])
# convert imagem para um array numpy
img_array = img_to_array(img).astype(np.uint8)
img_array = img_array.reshape((28, 28))
#conversão para um array numpy
lista = np.array([img_array])
testes = modelo.predict(lista)
print('resultado teste:', np.argmax(testes[0]))
Espero ter ajudado, qualquer duvida, não hesite em perguntar.
Bons Estudos