ImageAI-口罩辨識練習
ImageAI官網有附custom object detection的範例, 相信因該蠻多人都希望可 該範例跑自己想要的物件辨識.最近剛好遇到"covid-19"疫情.就以口罩配戴為例進行練習.
系統架構及概念:
".h5"和"json檔"為依照蒐集到的樣本學習完成後的YoloV3描述檔和設定檔,以我們的應用來說,要學習的目標為"有戴口罩的臉"和"沒戴口罩的臉".詳細的樣本製作和學習方式,我們會再另外再討論.這邊就以我先訓練好的檔案為例.
程式範例及解說我放在google colab中,大家可以自行在上面測試範例和修改,(順便幫我改改英文和國文~~).裡面也有附已經訓練好的YoloV3檔案.
google colab連結:
https://colab.research.google.com/drive/1NB-_CHR-LOW7sRJ63DrH5kF1yepMHVjT?usp=sharing
關於樣本的蒐集:
https://madeinti.blogspot.com/2020/05/imageai_8.html
參考資料:
https://github.com/OlafenwaMoses/ImageAI/blob/master/imageai/Detection/Custom/CUSTOMDETECTION.md
留言
張貼留言