在Google Colab使用yolov4訓練是否戴口罩
需先安裝yolov4,請參考在Google Colab安裝yolov4進行訓練與開啟攝影機進行辨識
Step1)連結帳號的Google雲端硬碟,點選「檔案->掛接雲端硬碟」
Step2)從Kaggle下載是否戴口罩圖片進行訓練
(1)在雲端硬碟的darknet資料夾,建立mask子資料夾。
(2)從https://www.kaggle.com/datasets/aditya276/face-mask-dataset-yolo-format/download 下載口罩圖片,並解壓縮後上傳到資料夾mask。
Step3)建立口罩辨識的yolo設定檔obj.data與obj.names
Step4)建立暫存權重的資料夾
Step5)修改/darknet/cfg/yolov4-tiny.cfg
修改以下幾行
第6行:batch=24
第7行:subdivisions=8
第221行:filters=21
第229行:classes=2
第274行:filters=21
第280行:classes=2
說明:batch=24,每批次取24張圖片進行訓練,subdivisions=8,每批次拆成8組,filters=(classes+5)*3,分成戴口罩與未戴口罩,設定classes為2
Step6)開始訓練
Step7)開啟攝影機進行辨識,修改第5行的設定
辨識如下