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

#深度學習 #cnn #圖片辨識

ellieytc 2020-08-27 03:37:18 ‧ 1947 瀏覽

我用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邦好手 1 級 ‧ 2020-08-27 08:41:48 檢舉

單純第一個念頭是再增加第11類是無法辨識的類別
或是預測機率最高還是低於某個值，就變成無法辨識？

ellieytc iT邦新手 5 級 ‧ 2020-08-28 21:13:33 檢舉

增加第11類是無法識別好像可以，但是設定預測機率門檻就不知道怎麼寫了@@？感謝你，我再試試看～

登入發表討論

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

1 個回答

I code so I am

iT邦高手 1 級 ‧ 2020-08-28 10:57:34

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

output_layer = Dense(10, activation='softmax', name='softmax')(x)

回應 1
分享
檢舉

ellieytc iT邦新手 5 級 ‧ 2020-08-28 21:14:06 檢舉

好的，謝謝你！

登入發表回應

我要發表回答

立即登入回答