HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Subpixel Heatmap Regression for Facial Landmark Localization

Adrian Bulat Enrique Sanchez Georgios Tzimiropoulos

Subpixel Heatmap Regression for Facial Landmark Localization

Abstract

Deep Learning models based on heatmap regression have revolutionized the task of facial landmark localization with existing models working robustly under large poses, non-uniform illumination and shadows, occlusions and self-occlusions, low resolution and blur. However, despite their wide adoption, heatmap regression approaches suffer from discretization-induced errors related to both the heatmap encoding and decoding process. In this work we show that these errors have a surprisingly large negative impact on facial alignment accuracy. To alleviate this problem, we propose a new approach for the heatmap encoding and decoding process by leveraging the underlying continuous distribution. To take full advantage of the newly proposed encoding-decoding mechanism, we also introduce a Siamese-based training that enforces heatmap consistency across various geometric image transformations. Our approach offers noticeable gains across multiple datasets setting a new state-of-the-art result in facial landmark localization. Code alongside the pretrained models will be made available at https://www.adrianbulat.com/face-alignment

Benchmarks

BenchmarkMethodologyMetrics
face-alignment-on-300wSHR-FAN
NME_inter-ocular (%, Challenge): 4.13
NME_inter-ocular (%, Common): 2.61
NME_inter-ocular (%, Full): 2.94
face-alignment-on-300w-split-2-300w-lpSH-FAN
AUC@7 (bbox): 71.1
NME (bbox): 2.04
NME (inter-ocular): 2.94
face-alignment-on-aflw-19SHR-FAN
AUC_box@0.07 (%, Full): 70.0
NME_box (%, Full): 2.14
NME_diag (%, Frontal): 1.12
NME_diag (%, Full): 1.31
face-alignment-on-cofwSH-FAN
NME (inter-ocular): 3.02%
face-alignment-on-cofw-68-300wlpSH-FAN
AUC@7: 64.9
NME (box): 2.47
face-alignment-on-wfw-extra-dataSH-FAN
AUC@10 (inter-ocular): 63.1
FR@10 (inter-ocular): 1.55
NME (inter-ocular): 3.72

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp