iT邦幫忙

1

[深度學習]CNN圖片分類模型的預測問題

我用Transfer Learning+CNN建了一個圖片分類模型,想要辨識10種食材:
apple(0), cabbage(1), carrot(2), chicken(3), cucumber(4), egg(5), mushroom(6), potato(7), radish(8), tomato(9)

我找了一個不在以上10種分類的圖片(花椰菜),丟進模型做預測,結果程式判斷它(花椰菜)是egg:
Imgur

想請教要怎麼讓程式跑出: 花椰菜不在以上分類中?

附上 完整程式碼

謝謝

hokou iT邦新手 4 級 ‧ 2020-08-27 08:41:48 檢舉
單純第一個念頭是再增加第11類是無法辨識的類別
或是預測機率最高還是低於某個值,就變成無法辨識?
ellieytc iT邦新手 5 級 ‧ 2020-08-28 21:13:33 檢舉
增加第11類是無法識別好像可以,但是設定預測機率門檻就不知道怎麼寫了@@? 感謝你,我再試試看~

1 個回答

0
I code so I am
iT邦研究生 3 級 ‧ 2020-08-28 10:57:34

下面這一行,10改成11就可以了

output_layer = Dense(10, activation='softmax', name='softmax')(x)
ellieytc iT邦新手 5 級 ‧ 2020-08-28 21:14:06 檢舉

好的,謝謝你!

我要發表回答

立即登入回答