【開發日記】馬桶識別之馬桶分類,利用百度人工智能定製化圖像識別進行分類

閒來沒事,經常逛百度的人工智能頁面,看有沒有上線什麼新功能。

偶然看到百度人工智能的圖片識別模塊裏面有一個“定製化圖片識別”模塊,可以上傳自己收集的圖片,然後百度後臺進行分類。有這等好事,當然是試一試了。剛好可以用我收集的不同馬桶型號的圖片去試一下這個模塊的性能。

這裏你唯一要做的是收集數據,並按照下圖的格式組織好數據就可以了。文檔中建議每類至少一百張。


準備好之後,壓縮成zip格式,創建模型,填寫一些描述,上傳數據即可。

訓練時間大概要兩三個小時吧,最後的訓練精度是72.91%。

這個精度比利用Tensorflow的遷移學習要高一點,但是對我來說,還不是很理想。

在這個模塊裏面還有有趣的功能,可以上線你訓練好的模型,而且可以進行校驗。

校驗模型的頁面如下, 你可以拖一張圖片進去,它可以判斷圖片屬於哪一類。相當於百度爲你的模型做了個Demo。


另外,如果你申請上線(差不多要4-5天時間),那麼可以通過調用API接口,進行目標識別。


是不是很好玩啊。


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章