SFRA: spatial fusion regression augmentation network for facial landmark detection
Cheng Peng,Guo Dong Li,Kun Zou,Bo Wen Zhang,Sio Long Lo,Ah Chung Tsoi
DOI: https://doi.org/10.1007/s00530-024-01510-w
IF: 3.9
2024-10-01
Multimedia Systems
Abstract:In this paper, we introduce a novel method for detecting facial landmarks, termed the spatial fusion regression augmentation network (SFRA), which aims to improve the spatial generalization capabilities of coordinate based regression techniques. The SFRA consists of three components, namely, the landmark feature enhancement network (LFE), spatial relationship modeling network (SRM), and landmark coordinate regression (LCR) network. The LFE is capable of incorporating prior information into the extracted landmark features, thus boosting their representational capacity. The SRM is capable of understanding the local interaction and global interaction among the landmark patches before they are passed to the LCR to predict the coordinates of the landmarks. Then, the output from each predicted coordinates from the LCR feeds into the subsequent stage's LFE to improve feature representation, creating a pyramid-like multistage structure that refines details progressively from a coarse to a fine level, ultimately achieving precise facial landmark predictions. The proposed method was rigorously tested on the 300W, WFLW and COFW datasets, achieving normalized mean error (NME) metrics of 3.07%, 4.11% and 3.29% respectively, thus placing it to be the best performing coordinate based regression methods; its performance when compared with the best heatmap based regression methods, is mixed, in that on the WFLW dataset, it is better, on the 300W dataset it is slightly worse, and on the COFW dataset, it is better if the normalization factor in the NME metric is based on inter-ocular distances, while using inter-pupil distances, it is slightly worse than the best method in the heatmap based regression methods. Lastly, ablation studies are performed to confirm the efficacy of the proposed approach.
computer science, information systems, theory & methods