前提・実現したいこと
勉強の一環として、犬猫を判別するプログラムを作成しました。
精度70%ほどの学習プログラムができたので、予測を行おうと思ったのですが上手くいきません。
適当に猫の画像を読み込ませて、それを正しく結果を表示させたいです。
発生している問題・エラーメッセージ
TypeError Traceback (most recent call last) <ipython-input-29-a65fe2ae938e> in <module> 29 30 for i, accuracy in enumerate(result): ---> 31 print('画像認識AIは「', folder[i], '」の確率を', np.int(accuracy * 100), '% と予測しました。') 32 33 print('-------------------------------------------------------') TypeError: only size-1 arrays can be converted to Python scalars
該当のソースコード
Python
1import cv2 2import matplotlib.pyplot as plt 3import numpy as np 4from keras.models import model_from_json 5 6 7recognise_image = './test_1/cat/17.jpg' 8 9folder = ['dog:犬', 'cat:猫'] 10 11image_size = 28 12 13color_setting = 3 14 15model = model_from_json(open('cnn_model.json', 'r').read()) 16model.load_weights('cnn_weights.h5') 17 18img = cv2.imread(recognise_image, 1) 19img = cv2.resize(img, (image_size, image_size)) 20plt.imshow(img) 21plt.show() 22 23img = img.reshape(image_size, image_size, color_setting).astype('float32')/255 24 25prediction = model.predict(np.array([img]), batch_size=2, verbose=1) 26result = prediction[0] 27 28for i, accuracy in enumerate(result): 29 print('画像認識AIは「', folder[i], '」の確率を', int(accuracy * 100), '% と予測しました。') 30 31print('-------------------------------------------------------') 32print('画像認識AI:犬 or 猫 の予測結果は、「', folder[result.argmax()],'」です。')
試したこと
調べてもよく分かりませんでした。。
補足情報
何か不足している情報があれば追記します。ご教示のほどよろしくお願いいたします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/04/09 05:03
2020/04/09 07:43