誰でも私を助けてくれます。私はKerasを使用してCNNにイメージのビルドバイナリ分類モデルを成功し、model.predict_classes()
を使用して予測を行い、ここに私のコードでいますケラでCNNを使用した画像のバイナリ分類で正しく予測された合計
import numpy as np
import os,sys
from keras.models import load_model
import PIL
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
model = load_model('./potholes16_2.h5')
model.compile (loss = 'binary_crossentropy',
optimizer = 'adam',
metric = ['accuracy'])
path= os.path.abspath("./potholes14/test/positive")
extensions = 'JPG'
if __name__ == "__main__":
for f in os.listdir(path):
if os.path.isfile(os.path.join(path,f)):
f_text, f_ext= os.path.splitext(f)
f_ext= f_ext[1:].upper()
if f_ext in extensions:
print (f)`enter code here`
img = Image.open(os.path.join(path,f))
new_width = 200
new_height = 200
img = img.resize((new_width, new_height), Image.ANTIALIAS)
#width, height= image.size
img = np.reshape(img,[1,new_width,new_height,3])
classes = model.predict_classes(img)
print (classes)
今、私は正確に予測画像の合計をカウントしたい、例えばどのように多くのクラスはクラス0またはクラス1に属していますか?
ありがとうございます。あなたはmodel.evaluate
機能を呼び出すために必要
実行推論をチェックする別の方法形状
(number_of_sample,1)
あるassums最後の行を、書くことでした。各バッチに対して、出力を配列/リストに挿入します。最終的な配列の混乱行列を計算してください – Nain
答えをありがとう、私は本当にそれを感謝し、私はそれを試してみます。 – Vosco