HRNet面部特征点检测

基于Heatmap的人脸特征点检测

HRNet面部特征点检测

基于TensorFlow 2的HRNet人脸特征点检测。

眼见为实,这里有一段视频DEMO:

HRNet人脸特征点检测_哔哩哔哩 (゜-゜)つロ 干杯~-bilibili
HRNet%E4%BA%BA%E8%84%B8%E7%89%B9%E5%BE%81%E7%82%B9%E6%A3%80%E6%B5%8B%E7%9A%84TensorFlow%E5%AE%9E%E7%8E%B0%E3%80%82%E6%BA%90%E7%A0%81%E5%BC%80%E6%94%BE%EF%BC%8C%E5%8F%AF%E4%BB%A5%E8%87%AA%E8%A1%8C%E6%94%B9%E8%BF%9B%E6%A8%A1%E5%9E%8B%E3%80%82%0A%E6%BA%90%E7%A0%81%EF%BC%9Ahttps://github.com/yinguobing/facial-landmark-detection-hrnet%0ABGM%EF%BC%9ARauf%20&%20Faik%20%D1%8D%D1%82%D0%BE%20%D0%BB%D0%B8%20%D1%81%D1%87%D0%B0%D1%81%D1%82%D1%8C%D0%B5?

亮点

  • 基于TensorFlow 2的Keras API
  • 支持多种数据集:WFLW, IBUG等
  • 先进的网络架构:HRNet v2
  • 数据增强:随机缩放、翻转、旋转
  • 模型优化:量化、剪枝‌‌‌‌

受限于硬件条件,Demo使用的模型是在batch_size=16, epochs=60的情况下得到的。如果你的硬件条件好,可以尝试训练一个更好的版本。

源代码

yinguobing/facial-landmark-detection-hrnet
A TensorFlow implementation of HRNet for facial landmark detection. - yinguobing/facial-landmark-detection-hrnet

欢迎star/fork/issue~