使用pytroch實現的基於yolov3的輕量級人臉檢測

使用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


使用pytroch實現的基於yolov3的輕量級人臉檢測


使用pytroch實現的基於yolov3的輕量級人臉檢測



分享到:


相關文章: