3 回答

TA贡献1836条经验 获得超5个赞
from keras.models import load_model
from keras.preprocessing import image
model=load_model("/blah/blah/blah")
img = image.load_img(path, color_mode = "grayscale", target_size=(128, 128, 1))
y = image.img_to_array(img)
y = np.expand_dims(y, axis=0)
images = np.vstack([y])
classes = model.predict(images/255.0, batch_size=8, verbose=0)

TA贡献1821条经验 获得超6个赞
predict 返回一个包含预测的列表。你可以用这个
results = model.predict(data)
for result in results:
print(str(result))
这将返回
0.99
0.87
0.75
或者如果你在另一个列表中有这些类,你应该这样做。
res = model.predict(data)
results = [[i,r] for i,r in enumerate(res)]
results.sort(key=lambda x: x[1], reverse=True)
for r in results:
print(classes[r[0]], str(r[1])))
这返回
("classA", 0.99)
("classB", 0.95)
添加回答
举报