#!/usr/local/bin/python3
#!-*- coding: utf-8 -*-

import os
import numpy as np
from keras.models import model_from_json
from keras.preprocessing.image import load_img, img_to_array

if __name__ == '__main__':

    #Chargez le modèle
    model_json = open('model.json').read()
    model = model_from_json(model_json)
    model.load_weights('model.h5')

    #Charger l'image
    image = load_img('{Chemin du fichier image}', target_size=(32, 32))

    #Convertir en tableau
    x = img_to_array(image)

    #Augmenter le nombre de dimensions
    x = np.expand_dims(x, axis=0)

    #Les données d'entrée sont[0,1]Normalisé à la plage de
    x = x.astype('float32')
    x /= 255.0

    #Prévoir
    preds = model.predict(x)

    #Obtenir l'index de la valeur maximale
    answer = np.argmax(preds)

    cifar_map = {
        0: "airplane",
        1: "automobile",
        2: "bird",
        3: "cat",
        4: "deer",
        5: "dog",
        6: "frog",
        7: "hourse",
        8: "ship",
        9: "truck"
    }

    print(cifar_map[answer])

automobile

frog

hourse

ship

hourse

Recommended Posts