HRNet面部特征点检测
基于Heatmap的人脸特征点检测
基于TensorFlow 2的HRNet人脸特征点检测。
眼见为实,这里有一段视频DEMO:
亮点
- 基于TensorFlow 2的Keras API
- 支持多种数据集:WFLW, IBUG等
- 先进的网络架构:HRNet v2
- 数据增强:随机缩放、翻转、旋转
- 模型优化:量化、剪枝
受限于硬件条件,Demo使用的模型是在batch_size=16, epochs=60的情况下得到的。如果你的硬件条件好,可以尝试训练一个更好的版本。
源代码
欢迎star/fork/issue~
Comments ()