code-cloudsg / h3r

Heatmap Regression via Randomized Rounding

Heatmap Regression via Randomized Rounding

---

[2020/09/02]: the paper is available on ArXiv.

[2020/10/28]: add new models trained on LaPa dataset (106 facial landmarks).

Introduction

This repo contains the facial landmark detection code for "Heatmap Regression via Randomized Rounding".

Demo

export PYTHONPATH=./:$PYTHONPATH
python examples/demo.py --image data/demo.jpg --model models/wflw/hrnet18_256x256_p2/

Test on WFLW (98 facial landmarks)

python examples/test_wflw.py --model models/wflw/hrnet18_256x256_p1/

Backbone	BBox	Resolution	#Params	FLOPs	NME (%)
HRNet-W18	P1	256x256	9.69M	4.84G	3.81
HRNet-W18	P2	256x256	9.69M	4.84G	3.95
MobileNetV2	P2	256x256	0.60M	0.51G	4.45
MobileNetV2	P2	160x160	0.60M	0.20G	4.58
MobileNetV2	P2	128x128	0.60M	0.13G	4.72

Test on LaPa (106 facial landmarks)

python examples/test_lapa.py --model models/lapa/hrnet18_256x256_p2/

Backbone	BBox	Resolution	#Params	FLOPs	NME (%)
HRNet-W18	P2	256x256	9.69M	4.86G	1.40
MobileNetV2	P2	256x256	0.60M	0.52G	1.69
MobileNetV2	P2	128x128	0.60M	0.13G	2.08

Citation

@article{yu2020heatmap,
  title={Heatmap Regression via Randomized Rounding},
  author={Yu, Baosheng and Tao, Dacheng},
  journal={arXiv preprint arXiv:2009.00225},
  year={2020}
}

Contact

Baosheng Yu, [email protected].

Acknowledgement

https://github.com/HRNet