我已经使用keras创建了一个图像分类器,我正在尝试将model.predict()
用于单个图像。我的代码如下所示:
path = 's.jpg'
from keras.preprocessing import image
img_width, img_height = 224,224
img = image.load_img(path, target_size = (img_width, img_height))
img = image.img_to_array(img)
img = np.expand_dims(img, axis = 0)
a =model.predict(img)
但是带有概率分数的输出数组"a"始终按降序排列。 你知道为什么吗??我有三类。
预测返回类概率,[class1_prob, class2_prob, ....]
假设模型中有softmax
。
它不应该总是递减,除非你传递这样的数据,以便类概率总是按递减顺序递减。