Update README.md

xialuxi · Jan 9, 2021 · 50fe6ed · 50fe6ed
1 parent ecc79b5
commit 50fe6ed
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -1,4 +1,16 @@
 # yolov5_face_landmark
 基于yolov5的人脸检测，带关键点检测
 
-1在yolov5的检测基础上，加上关键点回归分支
+1，在yolov5的检测基础上，加上关键点回归分支
+
+2，detect_one.py是单张图片的测试代码，　基于部分wideface训练的模型，稍后在百度云公开。
+
+3，主要修改代码部分：
+
+(１）hyp.scatch.yaml中增加关键点loss的超参数
+
+(2)　yolo.py中增加了关键点回归的计算
+
+(3)　face_datasets.py为人脸数据的读取方式，准备数据的格式参考yolov5的格式，在后面增加关键点的左边（归一化）
+
+(4)　loss.py中增加关键点回归的loss计算