Data Science and Machine Learning - #45 Keras - Classificação de Imagens II

Neste tutorial iremos aprender como realizar a predição de imagens utilizando o módulo Keras do Python.

Predizendo imagens - Keras Python

Como o modelo de imagens do exemplo são imagens de 28px x 28px na cor preta, precisamos primeiramente pegar as imagens reais e transformá-las para este formato, e depois sim verificar a classificação delas:

def load_image(filename):
    img = load_img(filename,color_mode="grayscale",target_size=(28,28))
    img = img_to_array(img)
    img = img.reshape(1,28,28,1)
    img = img.astype('float32')
    img = img / 255.0
    return img

Plotando a imagem

A imagem ficará no seguinte formato:

newImg = load_image("new5.png")
newImg2 = newImg.reshape(28,28)
plt.figure()
plt.imshow(newImg2)
plt.colorbar()
plt.grid(False)
plt.show()

Realizando Predições de Imagens

Para realizar a predição da imagem, verificando se as fotos reais estão enquadrando nos grupos corretos, faremos o seguinte código:

pred = model.predict(newImg)
pred

np.argmax(pred)

class_names[0]

Por hoje é só! Sucesso nos códigos e na vida!