学習モデル hdf5ファイルを利用して画像認識して答えを算出

Question

画像認識をするためデータを集め学習させ推論するためのモジュールをhdf5ファイルに
保存しました。

そのhdf5ファイルを利用して実際に画像データから文字を読み取り記載文字を算出するコードを
作成しました。
問題：
文字を認識した可能性の高い算出値が何番目にあるかは出せましたが、具体的に画像から読み取れた
文字が何なのかがうまく引き出せません。

どのようにすれば画像から読み取った文字を引き出だせるものかご教示お願いします。

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

jpg_name = 'img'
model_file_name='cnn_002'

model=load_model('./hdf5/' + model_file_name+'.hdf5')
# target_size=(80,28)
img_path = ('./img/' + jpg_name + '.jpg')

#　実際の画像をロードします。
img = img_to_array(load_img(img_path, target_size=(80,28)))

#正規化処理
img=np.array(img, dtype=np.float)/255
img = img.reshape([-1, 28, 80, 3])
#　実際に予測する
pred = model.predict(img, batch_size=1, verbose=0)
pred_label = np.argmax(pred[0])

print('name00:',pred)
print('name01:',pred_label)
print('name02:',pred[0][pred_label])

printではそれぞれ
name00: [[6.5707992e-04 4.1942326e-06 3.3447006e-05 ... 1.4371148e-03
2.4168508e-03 1.9060132e-03]]
name01: 2172
name02: 0.008105861

となります。

Answer 1

np.argmax()は引数にaxisを設定しないと、配列全体のなかの最大を取得します。やりたいことは画像単位で最大値を取得することと思うので、axisパラメーター(確信はありませんが、axis=1)を必要があります。

Comments

[matrix99999]

R.Shigemori様
ありがとうございます。
画像認識した最大値のラベル（予測された画像の文字）を取得したいと考えています。
pred_label = np.argmax(pred,axis=1)で実行しましたが
たぶん配列の最大値を取得したと思います。