本篇的人臉偵測使用 Darknet 進行訓練,model 為 YOLOv4-tiny
訓練集、驗證集使用 WIDER FACE
執行 data_preprocessing.ipynb,其中 WIDER_FACE_covert_to_yolo.py 用於轉換資料集格式,記得修改檔案路徑
若要使用較少資料量的 WIDER FACE:training data 4000、validation data 1000 筆,可從下列連結中拿取
yolo_train: Google-Drive
yolo_val: Google-Drive
訓練步驟都寫在 yolov4-tiny.ipynb,可在 Colab 上進行訓練
訓練完的模型放置在 weights
在 Colab 頁面按 F12,在 Console 貼上以下 Code 後按 Enter
function ConnectButton(){
console.log("Connect pushed");
document.querySelector("#connect").click()
}
setInterval(ConnectButton,60000);
使用 opencv 進行 inference
$ python face_detect.py
詳細流程可以參考我的 Medium: https://medium.com/@chingi071/yolo-c49f70241aa7