簡體   English   中英

Keras:解釋model.predict()進行圖像分類

[英]Keras: Interpretation of model.predict() for image classification

我已經使用 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)

但是具有概率分數的 output 數組“a”始終按降序排列。 你知道為什么嗎? 我分為三類。

假設您的 model 中有softmax ,預測返回 class 概率, [class1_prob, class2_prob, ....]

它不應該總是遞減,除非你傳遞這樣的數據,這樣 class 的概率總是按遞減順序排列。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM