[問題] CNN辨識特定物件

作者: gs8613789 (Shang6029)   2017-09-05 14:25:33
大家好,我是深度學習的新手,因為PTT似乎沒有深度學習相關的板,所以發在這裡。
最近在看CNN,知道他預測的結果是一連串的label,而值就是測試圖片為哪一個label的
機率。
那假如說我想要建立一個只有兩個label的模型,值就是YES或NO
比方說我想要辨識一張椅子,我想要知道的就只有他是椅子的機率。
當我測試圖片為一張椅子時,他就跟我說這是一張椅子。
當我測試圖片為其他東西(比方說一個人或一個水壺)的時候,他就跟我說這不是一張椅子

請問這樣子模型要如何訓練呢?爬了許多文找不到相關的討論,懇請各位給我關鍵字。謝
謝!
作者: weihan0405 (blue)   2017-09-05 15:38:00
很容易啊,給它training data 椅子的圖label都設1 其他不是椅子圖全部設2去訓練,最後的layer 參數記得給兩類,這樣應該就可行了
作者: indigolemon (藍色拉拉喵)   2017-09-05 15:46:00
作者: frank910138 (frank)   2017-09-05 16:41:00
借問一下..training 資料量大概要多少?
作者: EGsux (天龍人)   2017-09-06 06:56:00
跟你model複雜度有關 你也可以用pre-training network最後一兩個layer drop掉加新的去trainimage可以去image net
作者: bbkingck (Twister)   2017-09-06 15:59:00
另外,建議非正解的東西設成0(或是-1),表示機率較直觀

Links booklink

Contact Us: admin [ a t ] ucptt.com