-
基于yolov3的轻量级人脸检测模型(包含关键点)以pytorch实现,被称为yoloface大礼包
资源介绍
yolo-face-with-landmark
实现的功能
添加关键点检测分支,使用wing loss
Installation
Clone and install
git clone
使用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
demo
pytho