閒來沒事,經常逛百度的人工智能頁面,看有沒有上線什麼新功能。
偶然看到百度人工智能的圖片識別模塊裏面有一個“定製化圖片識別”模塊,可以上傳自己收集的圖片,然後百度後臺進行分類。有這等好事,當然是試一試了。剛好可以用我收集的不同馬桶型號的圖片去試一下這個模塊的性能。
這裏你唯一要做的是收集數據,並按照下圖的格式組織好數據就可以了。文檔中建議每類至少一百張。
準備好之後,壓縮成zip格式,創建模型,填寫一些描述,上傳數據即可。
訓練時間大概要兩三個小時吧,最後的訓練精度是72.91%。
這個精度比利用Tensorflow的遷移學習要高一點,但是對我來說,還不是很理想。
在這個模塊裏面還有有趣的功能,可以上線你訓練好的模型,而且可以進行校驗。
校驗模型的頁面如下, 你可以拖一張圖片進去,它可以判斷圖片屬於哪一類。相當於百度爲你的模型做了個Demo。
另外,如果你申請上線(差不多要4-5天時間),那麼可以通過調用API接口,進行目標識別。
是不是很好玩啊。