Skip to content

xialuxi/yolov5_face_landmark

Repository files navigation

yolov5_face_landmark

基于yolov5的人脸检测,带关键点检测

1,在yolov5的检测基础上,加上关键点回归分支

2,detect_one.py是单张图片的测试代码, 基于部分wideface训练的模型,稍后在百度云公开。

3,主要修改代码部分:

(1)hyp.scatch.yaml中增加关键点loss的超参数

(2) yolo.py中增加了关键点回归的计算

(3) face_datasets.py为人脸数据的读取方式,准备数据的格式参考yolov5的格式,在后面增加关键点的左边(归一化)

(4) loss.py中增加关键点回归的loss计算

About

基于yolov5的人脸检测,带关键点检测

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published