iT邦幫忙

2022 iThome 鐵人賽

DAY 27
2
自我挑戰組

PixelBit 可以這樣玩!系列 第 27

(Day 27)訓練 Image Classification影像分類 AI 模型

  • 分享至 

  • xImage
  •  

昨天我們已經註冊好帳號,並且啟用學生版帳戶,也建立了影像分類的專案,今天我們將開始蒐集資料、訓練模型,最後測試模型。

蒐集圖片資料

在訓練模型之前我們需要先蒐集圖片資料,這裡我們使用 Teachable Machine 工具幫我們擷取影像為圖片。

我們將蒐集三種圖片

  • Dinosaur x 300 張
  • Ghost x 300 張
  • Shark x 300 張

https://ithelp.ithome.com.tw/upload/images/20221008/201520659d115A1aLV.png

官方建議是每個類最好要有50張圖片以上

上傳圖片

強烈建議使用 Google Chrome 操作,避免異常

打開昨天建立的影像分類專案。

https://ithelp.ithome.com.tw/upload/images/20221008/20152065i5vMYbWz31.png

開始上傳圖片

https://ithelp.ithome.com.tw/upload/images/20221008/20152065GjFSLDZl7j.png

輸入標圖片標籤名稱,按下上傳

https://ithelp.ithome.com.tw/upload/images/20221008/20152065JVoeA3JyQT.png

上傳完畢後系統還會自動檢查重複的照片並將其移除,以這次的例子有14張圖片被判定為重複

https://ithelp.ithome.com.tw/upload/images/20221008/20152065awFjooUKeJ.png

接著我們繼續新增其他圖片,直到所有類別都完成

https://ithelp.ithome.com.tw/upload/images/20221008/20152065teQxXJdwVH.png

https://ithelp.ithome.com.tw/upload/images/20221008/20152065KrsRdar5o3.png

https://ithelp.ithome.com.tw/upload/images/20221008/20152065F7585Zrr8K.png

訓練模型

選擇訓練

https://ithelp.ithome.com.tw/upload/images/20221008/20152065BpWMKHgcqW.png

選擇快速訓練即可

https://ithelp.ithome.com.tw/upload/images/20221008/20152065NAIXPQGuv7.png

等待訓練完成

https://ithelp.ithome.com.tw/upload/images/20221008/20152065MFH0gGklLh.png

評估模型

訓練完成將會顯示訓練結果,可以看到整體與各別的評估數值

  • Precision:如果你的模型預測了一個標籤,那麼它正確的可能性有多大?
  • Recall:在應該正確預測的標籤中,你的模型正確找到的百分比是多少?
  • AP:模型性能的度量,總結了不同閾值的精度和召回率。

https://ithelp.ithome.com.tw/upload/images/20221008/20152065GMHzhQus7S.png

使用雲端平台測試模型

接著我們可以透過平台內建的測試工具實際測試模型效果,選擇右上方快速測試

https://ithelp.ithome.com.tw/upload/images/20221008/20152065IxG4JXbq6t.png

上傳圖片後將會開始預測,並將各個類的可能性顯示出來,可以看到這張圖有 99.9%是恐龍

https://ithelp.ithome.com.tw/upload/images/20221008/20152065Mbs9Y0L88N.png

發布模型並使用 REST API 測試模型

點選發布,設定模型名稱以及預測資源

https://ithelp.ithome.com.tw/upload/images/20221008/201520650ogohTY9Uc.png

成功發布模型後,你會在左側欄看到已發布的標籤

https://ithelp.ithome.com.tw/upload/images/20221008/20152065zCjSVe0q1N.png

透過上方預測 URL 取得預測網址以及參數說明,可透過圖片網址或圖片檔案呼叫 API

https://ithelp.ithome.com.tw/upload/images/20221021/20152065xvNo8HCR1h.png

使用 Postman測試預測 API,依照上方說明設定 Header、Body

https://ithelp.ithome.com.tw/upload/images/20221008/20152065g7sghXjW9W.png

https://ithelp.ithome.com.tw/upload/images/20221008/20152065zAkZRQLxjb.png

呼叫預測 API 取得圖片預測結果,可以看到此圖有 99.9% 是恐龍

https://ithelp.ithome.com.tw/upload/images/20221008/20152065K3K1RvH743.png

小結

以上就是整個 Azure Cutom Vision 影像分類的過程,我們只需要蒐集圖片,上傳到此平台,即可透過平台幫我們整理資料及訓練模型,最後還有評估及測試的功能,預測 REST API 也算好用,預測耗費時間平均為1秒,還算不錯,整體來說算是做得相當完善了,對於任何初學者都會非常容易上手。

更多有趣系列教學文章


上一篇
(Day 26)Azure Cutom Vision 註冊帳號與專案準備
下一篇
(Day 28)訓練 Object Deteion 物件偵測 AI 模型
系列文
PixelBit 可以這樣玩!30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言