Magicode logo
Magicode
2 min read

AIにアミバとトキを学習させよう⑤*オチあり

https://cdn.apollon.ai/media/notebox/8803aa0b-c9ba-4cab-9807-2580922d9c5e.jpeg
前回つくったモデルを読み込んで予測してみます。 使う木人形(デク)はこいつです。
さて結果は、、、
あれ、、、、?
ひ、暇ができたら学習データのクレンジングとか選定のやり直しでもっかいチャレンジしたいとおもいます。
うわらば、、、。
import keras
import sys, os
import numpy as np
from PIL import Image
from keras.models import load_model

imsize = (64, 64)
testpic     = "./test.jpg"
keras_param = "./cnn.h5"
def load_image(path):
    img = Image.open(path)
    img = img.convert('RGB')
    img = img.resize(imsize)
    # 画像データをnumpy配列の形式に変更
    img = np.asarray(img)
    img = img / 255.0
    return img
model = load_model(keras_param)
img = load_image(testpic)
prd = model.predict(np.array([img]))
print(prd) 
prelabel = np.argmax(prd, axis=1)
if prelabel == 0:
    print(">>> アミバ様")
elif prelabel == 1:
    print(">>> トキ")

Discussion

コメントにはログインが必要です。