使用pytroch實現的基於yolov3的輕量級人臉檢測
yolo-face-with-landmark是一個使用pytroch實現的,基於yolov3的輕量級人臉檢測。
包含了關鍵點檢測。
安裝:
下載代碼。
使用src/retinaface2yololandmark.py腳本將retinaface的標記文件轉為yolo的格式使用, 使用src/create_train.py 創建訓練樣本。
訓練 python train.py --net mbv3_large_75 --backbone_weights \ ./pretrained/mobilenetv3-large-0.75-9632d2a8.pth --batch-size 16
測試 python evaluation_on_widerface.py cd widerface_evaluate python evaluation.py
演示 python demo.py