在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行的設定

辨識如下