基于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计算