chingi071 / ScaledYolov4_FaceMaskDetection

11 stars 4 forks source link

ScaledYolov4_FaceMaskDetection

本篇使用 Scaled-YOLOv4 進行訓練,model 為 YOLOv4-csp

訓練集、驗證集使用 Kaggle 口罩資料集: https://www.kaggle.com/andrewmvd/face-mask-detection/metadata

轉換資料集格式

執行 xml_covert_to_yolo.py,記得修改檔案路徑

訓練模型

訓練步驟都寫在 yolov4-csp.ipynb,在 Colab 上進行訓練

檔案放置路徑如下

訓練流程

詳細流程可以參考我的 Medium: https://medium.com/@chingi071/scaledyolov4-yolov4-csp-%E8%A8%93%E7%B7%B4%E6%95%99%E5%AD%B8-ee091598e503