[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Next Article in Journal
High-Performance SAW Resonator with Spurious Mode Suppression Using Hexagonal Weighted Electrode Structure
Previous Article in Journal
CVII: Enhancing Interpretability in Intelligent Sensor Systems via Computer Vision Interpretability Index
You seem to have javascript disabled. Please note that many of the page functionalities won't work as expected without javascript enabled.
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Self-Attention Mechanism-Based Head Pose Estimation Network with Fusion of Point Cloud and Image Features

1
College of Civil Engineering, Hefei University of Technology, Hefei 230009, China
2
Key Laboratory of Geospatial Technology for the Middle and Lower Yellow River Regions, Henan University, Ministry of Education, Kaifeng 475004, China
*
Author to whom correspondence should be addressed.
Sensors 2023, 23(24), 9894; https://doi.org/10.3390/s23249894
Submission received: 15 November 2023 / Revised: 9 December 2023 / Accepted: 14 December 2023 / Published: 18 December 2023
(This article belongs to the Section Physical Sensors)
Figure 1
<p>The top is the RGB image, and the bottom is the head pose label corresponding to the RGB image. (<b>a</b>) When the yaw angle is 38.27°, the discretized results of the head pose. (<b>b</b>) When the yaw angle is −40.39°, the discretized results of the head pose.</p> ">
Figure 2
<p>Head pose estimation network. Different colors represent the characteristics of different stages. The network is mainly divided into four modules: feature function, fusion function, score function, and predict function.</p> ">
Figure 3
<p>Feedforward residual MLP module.</p> ">
Figure 4
<p>Spatial self-attention module.</p> ">
Figure 5
<p>Feature extraction module. The network consists of five residual point blocks, three attention blocks, and a feature transform.</p> ">
Figure 6
<p>Classification and regression module.</p> ">
Figure 7
<p>Data source. (<b>a</b>) RGB image. The head position in the RGB image. (<b>b</b>) Head mask. In the head mask, the white region is the head, while the black region is the background. (<b>c</b>) Point cloud. From the depth map to the point cloud.</p> ">
Figure 8
<p>Point clouds at different scales. (<b>a</b>) Primary point cloud. (<b>b</b>) Downsampling the original point cloud to 1024. (<b>c</b>) Downsampling the original point cloud to 512.</p> ">
Figure 9
<p>When comparing model prediction accuracy, within the same dataset, between instances with and without positional encoding. (<b>a</b>) The predictive accuracy of the model on yaw angles. (<b>b</b>) The predictive accuracy of the model on pitch angles. (<b>c</b>) The predictive accuracy of the model on roll angles.</p> ">
Figure 10
<p>When using the 11th and 12th instances as the dataset, a comparison between the model’s predicted values and the ground truth. (<b>a</b>) Comparison on yaw angles. (<b>b</b>) Comparison on pitch angles. (<b>c</b>) Comparison on roll angles.</p> ">
Figure 11
<p>When each set of data in the dataset is used as a test set, the model’s predicted results are compared to the ground truth, and the average absolute error is calculated. The red dotted line represents the upper limit of the model’s prediction accuracy.</p> ">
Figure 12
<p>Comparison of different methods on the BIWI dataset. (<b>a</b>) The comparison of different methods in terms of yaw angle prediction accuracy. (<b>b</b>) The comparison of different methods in terms of pitch angle prediction accuracy. (<b>c</b>) The comparison of different methods in terms of roll angle prediction accuracy.</p> ">
Figure 13
<p>Visualization of partial test set results. The blue, green, and red colors respectively represent yaw, pitch, and roll angles. The top and bottom sections show the RGB and point cloud visualizations, respectively.</p> ">
Versions Notes

Abstract

:
Head pose estimation serves various applications, such as gaze estimation, fatigue-driven detection, and virtual reality. Nonetheless, achieving precise and efficient predictions remains challenging owing to the reliance on singular data sources. Therefore, this study introduces a technique involving multimodal feature fusion to elevate head pose estimation accuracy. The proposed method amalgamates data derived from diverse sources, including RGB and depth images, to construct a comprehensive three-dimensional representation of the head, commonly referred to as a point cloud. The noteworthy innovations of this method encompass a residual multilayer perceptron structure within PointNet, designed to tackle gradient-related challenges, along with spatial self-attention mechanisms aimed at noise reduction. The enhanced PointNet and ResNet networks are utilized to extract features from both point clouds and images. These extracted features undergo fusion. Furthermore, the incorporation of a scoring module strengthens robustness, particularly in scenarios involving facial occlusion. This is achieved by preserving features from the highest-scoring point cloud. Additionally, a prediction module is employed, combining classification and regression methodologies to accurately estimate head poses. The proposed method improves the accuracy and robustness of head pose estimation, especially in cases involving facial obstructions. These advancements are substantiated by experiments conducted using the BIWI dataset, demonstrating the superiority of this method over existing techniques.

1. Introduction

The accurate determination of head poses forms the foundation of human–computer interaction and facial analysis. Head pose estimation is widely applicable across numerous fields, such as line-of-sight estimation [1], human–computer interaction [2,3,4,5], and recognition of driving attention [6].
The estimation of head pose recognition can be divided into two categories based on the method used: head pose estimation with facial features and head pose estimation without facial features. Head pose estimation with feature points relies on determining a head’s orientation through the relative positions of facial feature points on the human face. In [7], researchers utilized OpenFace(2.2.0), an open-source software, to extract facial feature points and applied the Solve-Perspective-n-Point (SolvePnP) algorithm for head pose estimation. Experiments conducted on the BIWI dataset achieved absolute errors of 7.9°, 5.6°, and 4.5° for the yaw, roll, and pitch angles, respectively. However, the estimation of a head pose from 2D feature estimation yields lower accuracy owing to the necessity of 3D information. In [8], researchers employed a cascade regression tree model to localize facial feature points on the human body using a stereo camera setup to recover these feature points’ three-dimensional (3D) coordinates. Utilizing the 3D information of the feature points led to more accurate head pose estimations, particularly in the range from −15° to 15° of head rotation. Consequently, utilizing 3D information enhances the accuracy of head pose estimation compared to relying solely on two-dimensional (2D) feature points. While the use of binocular cameras has shown favorable results in small-scale head pose estimations, its accuracy diminishes considerably when applied to large-scale head pose estimations. Zhao et al. [9] proposed a head pose tracking method based on scale-invariant feature transform (SIFT) feature point matching. This method detects SIFT feature points in two consecutive frames and estimates the head pose by calculating a rotation matrix based on the 3D information of the matched feature points. This approach addresses the challenge of reduced estimation accuracy under large pose angles, which stems from imprecisions in feature point localization. However, error accumulation in estimating continuous head poses remains a concern. Liu et al. [10] reconstructed a 3D face model using facial key-point coordinates for head pose estimation, achieving an average accuracy of 7.05° on the BIWI dataset. Head pose estimation without feature points relies primarily on machine learning and deep learning methods [11,12,13,14], which demonstrate flexibility and robustness in handling occlusions and extreme pose variations. Ruiz et al. [15] introduced a novel network named HopeNet, which accepts RGB images as input, utilizes ResNet50 to extract image features, and transforms the regression problem into a classification problem, thereby addressing head pose estimation without landmarks. This convolutional neural network (CNN)-based head pose estimation approach demonstrated higher accuracy and stability than methods relying on feature points. However, this method’s dependency on ResNet50 as the backbone compromises its real-time applicability. Inspired by the soft stagewise regression network (SSRNet), Yang et al. [16] introduced the fine-grained structure aggregation network (FSANet), which surpassed the HopeNet network in terms of both speed and accuracy. While training the HopeNet network, definite losses computation using one-hot labels introduces significant errors when classifying two samples with slight differences. Advancements in deep learning have led to the preprocessing of data labels to enable models to capture high-frequency information from data. Label distribution learning was demonstrated to be effective in [17,18], employing soft labels generated using Gaussian functions instead of one-hot vectors for training head pose models [19]. These methods demonstrated improvements over the HopeNet network. However, based on the prediction results from HopeNet and FSANet, these methods struggle in scenarios involving changes in illumination and self-occlusion. The accessibility of sensors and the development of RGB-D cameras at an increasingly affordable rate have led to the integration of red, green, blue, and depth (RGB-D) images into the head pose estimation process, significantly enhancing accuracy. A head pose estimation method based on Kalman filtering and random forest regression was proposed [20]. This method utilizes depth information from a depth image to estimate head pose. Compared to RGB images, depth images contain more information, thus enhancing head pose estimation accuracy. Wang et al. [21] obtained promising results in head pose estimation by employing point cloud fusion and registration methods, further optimizing the process with a particle filter. Shihua et al. [22] utilized RGB images to convert depth maps into point clouds and proposed a new deep learning network called head pose estimation network (HPENet). This network accepts point cloud data as input and employs HPENet for feature extraction; however, this method exhibits significant estimation errors for extreme head pose angles. In [23], PointNet++ combined with graph CNNs (GCNNs) was employed as a feature extraction network to transform head pose estimation from a regression problem to a classification problem, addressing the issue of large estimation errors in significant head pose angles.
Many existing head pose estimation methods rely on a singular data source, typically utilizing either an RGB or depth image as the input to establish the mapping relationship between 2D and 3D spaces. However, these methods often fall short of delivering accurate or efficient predictions. Consequently, integrating multiple data sources with complementary characteristics becomes imperative for enhancing the overall perception capabilities. This study employed depth and RGB images to generate point cloud data representing the human head. Introducing spatial self-attention mechanisms into PointNet involved replacing the multilayer perceptrons (MLPs) within PointNet with feedforward residual MLPs, resulting in an enhanced PointNet network for extracting spatial features from the point cloud data. Additionally, the ResNet network concurrently extracted textural features. Subsequently, the postprocessing stage fused the features extracted from both networks. Each subnetwork responsible for head pose estimation included fully connected and SoftMax layers, sharing the combined features derived from both point cloud and image data. Moreover, Gumble functions were used to probabilistically encode head pose labels, while the network updated its parameters through a combination of Kullback–Leibler (KL) divergence and expected regression loss. To validate the method’s reliability, the BIWI depth and RGB images were utilized as training and testing datasets for the model. The experiments demonstrated that the proposed method achieved a high accuracy in estimating head poses, particularly when evaluated on the publicly available BIWI dataset.
The remainder of this paper follows the structure below. Section 2 introduces discretization methods and compares them with a Gaussian discretization method. Section 3 outlines the model, categorizing it into four parts and providing separate introductions for the feature extraction, feature fusion, feature scoring, and prediction modules. Section 4 introduces and deliberates on the dataset utilized for the experiments, followed by an in-depth exploration of the conducted experiments and their results. Finally, Section 5 offers a summary of the entire paper and presents conclusions derived from the discussed findings and results throughout the document.

2. Discretization of Head Pose Labels

Currently, two primary regression approaches exist. The first is direct regression, which may not accurately predict samples affected by noise. The second method involves regression with classification constraints. In the realm of image-based head pose recognition, Ruiz et al. employed a classification layer to comprehend the data distribution and discretize the regression space. They transformed the direct regression of head pose angles into a data distribution process, generating expected values within predefined classification intervals. The process of computing head pose angles included assigning weights to the predicted values based on their corresponding angles and calculating the regression loss. Subsequently, a combination of classification and regression weights was applied to adjust the contribution of different data distributions during network training. However, head pose angles are arbitrary and continuous within specific ranges. A classification-based approach for head pose estimation might introduce more significant errors, particularly when the input samples exhibit high correlations and low discriminative power. The application of a classification method may result in placing two closely related samples into the same class. To mitigate such issues, a study [24] introduced a label-wise distribution learning approach. This approach transforms a set of Euler angles x = { y , p , r } for head poses into discrete Gaussian distribution labels. Here, “ x = { y , p , r } ” denote the yaw angle, pitch angle, and roll angle of the head’s orientation, respectively. These Gaussian labels offer more comprehensive information compared to the original head pose labels, as they account for the actual scalar labels and relationships between neighboring elements within the accurate labels. However, the feature symmetry of Gaussian functions might not adequately capture visual differences in head pose images. To address this, a novel method for discretizing head pose labels was introduced in this study. This method involves both Gaussian and Gumbel functions [25], as illustrated in Equation (1), where β , s l , δ , and y represent the scale parameters, position parameters, variance of Gaussian functions, and head pose label values, respectively.
1 β × e x p ( e x p ( y s l β ) y s l β ) β = 3 ,   y > 0 e x p ( ( y s l ) 2 2 δ 2 ) δ = 1 ,   y = 0 1 β × e x p ( e x p ( y s l β ) y s l β ) β = 3 ,   y < 0
The discrete head pose label distributions for angles of 38.27° and −40.39° are illustrated in Figure 1, where the angle interval is 3°, and the values of δ and β are one and three, respectively.
As shown in Figure 1, the range of [−100, 100] is partitioned into 66 intervals with a spacing of 3, and the values of the current head pose are computed separately for each of these 66 intervals. In Figure 1a,b, it becomes evident that Gumbel-based discrete labels exhibit more logical consistency compared to Gaussian-based labels. For instance, visually, 38.27° and −40.39° align with a rightward and leftward tilt, respectively, aligning with the non-feature characteristics of the Gumbel distribution. This alignment reflects the intuitive angular intensity of the head pose in a visual context.
There exist various methods to describe the similarity between the two distributions. Following the network’s output through the SoftMax layer, this study utilized the KL divergence to quantify the similarity between the accurately observed and predicted label distributions, as shown in Equation (2).
T = k y k ln y ^ k
where y k , y ^ k , and k represent the accurate discrete distribution labels, the distribution predicted by the network, and the number of classification intervals, respectively.
Inspired by Ruiz et al., alongside measuring the distinction between label distributions, we implemented the expected regression to quantify the disparity between the actual and estimated head poses. This procedure involved computing the absolute difference between the product of the maximum component of the angular probability distribution and the corresponding angular interval representative value subtracted from the correct head pose angle, as depicted in Equation (3).
H = n = 1 k p k × y ^ k y
where y is the true head pose angle; y ^ k is the distribution predicted by the network, and p k is the representative value of the angle interval corresponding to each category in the predicted label distribution. Finally, the proposed head pose estimation loss function is defined as follows:
L = T + α × H
where α is a weighting parameter that balances the two components of the loss function.

3. Head Pose Estimation Network

The head pose estimation network proposed in this study comprises four principal components, as depicted in Figure 2. The algorithm employs a feature function to extract features from both image and point cloud data. Moreover, a fusion function is utilized to combine the extracted features obtained from the point cloud and image data. Additionally, the algorithm incorporates a score function responsible for assigning scores to each point cloud feature, ultimately selecting the feature with the highest probability value as the final feature. Finally, the algorithm uses a prediction function to estimate the head pose by considering the feature with the highest score.

3.1. Feature Function Module

While neural networks have proven effective in fitting nonlinear functions, this study identified a limitation in capturing high-frequency information within the point cloud when directly inputting the point cloud coordinates into the network for the head pose estimation task. As reported in [26], employing positional encoding functions to project inputs into a higher-dimensional space before feeding them into the network can enhance its capability to learn high-frequency features. Therefore, this study implemented the frequency-encoding method proposed in [27] to transform the coordinates of the point cloud input into a higher-dimensional space. Encoding Equation (5) is expressed as follows:
p ( x ) = ( sin ( 2 0 π x ) , cos ( 2 0 π x ) , . . . . , sin ( 2 L 1 π x ) , cos ( 2 L 1 π x ) )
where x presents a point within the point cloud coordinates, and L N * indicates the dimension of the positional encoding. The PointNet network, a deep learning architecture introduced by Charles et al. [28], is specifically designed to process unstructured point cloud data directly. This network incorporates a spatial transformer network comprising a multilayer perceptron to address issues related to translation and rotational invariance within the point clouds. In the model presented here, we incorporated the approach outlined in [29], replacing the MLP structure in the PointNet network with a feedforward residual MLP network to alleviate issues related to gradient explosion and vanishing caused by the depth of the network. Refer to Figure 3 for an illustration of the model’s structure.
Two MLPs primarily constitute this module, with the output of the last MLP layer added to the input data. This mechanism effectively addresses issues related to gradient explosion and gradients resulting from network depth. The core process is described by Equation (6).
y = F ( x , w i ) + x y = F ( x , w i ) + w x
where F ( x , w i ) represents the residual module to be learned, and w i and w represent the parameters for various MLP layers within the residual structure, encompassing nonlinear and linear transformations, respectively. These parameters correspond to the parameters for the linear mapping MLP, ensuring consistency in the input and output sizes in cases where the dimensions of the input data and MLP output differ.
In the conversion of depth maps into point cloud data, noise in the depth images often generates outliers and noisy points in the resulting point cloud. To tackle this issue, a spatial self-attention mechanism is integrated into PointNet to calculate the weight of each point. This integration reduces the influence of outliers and noisy point features within the network, subsequently enhancing the accuracy of head pose estimation. Figure 4 provides an illustration of the self-attention mechanism.
In Figure 4, N represents the number of point clouds, and D denotes the features for each point; the algorithm acquires the maximum and average values along the channels at each point. These two outcomes are then stacked together. A one-dimensional convolution is employed to align the number of channels to one. By utilizing a sigmoid function, the weights for each point are constrained within the range [0, 1]. Finally, the weight values are multiplied by the original values. Image features are extracted using a ResNet network. Figure 5 illustrates the comprehensive structure of the system.

3.2. Fusion Function Module

In [30], a method was proposed where features extracted through a traditional feature extractor (Gabor Filter) and those directly extracted using CNNs are fused in later stages. This approach effectively captures low-frequency and high-frequency information from data, leading to promising results in gesture recognition tasks. Additionally, in [31], the authors proposed a multi-level fusion module that integrates high-, medium-, and low-level features from RGB and thermal infrared images. This module effectively aggregates features, providing enhanced semantic information for urban scene segmentation. The aforementioned study highlights the efficacy of sensible feature fusion in efficiently aggregating features, enabling the network to acquire more information and, thus, enhancing the robustness of the model. Taking inspiration from [32,33,34], the point cloud features y 1 p o i n t and y 2 p o i n t y 3 p o i n t are concatenated. Subsequently, this combined feature is mapped to a higher-dimensional space through fully connected layers and is globally averaged to obtain a new global feature y 4 p o i n t . Simultaneously, the global features of the point cloud y 3 p o i n t and y 4 p o i n t , the global features of the image y 1 R G B , and the features of each point in the point cloud y 1 p o i n t and y 2 p o i n t y 3 p o i n t are concatenated to derive the final fused feature. This approach facilitates a comprehensive fusion of point clouds and image features, as the feature encompassed the global characteristics from each stage of the image and point cloud, along with the individual characteristics of each point in the point cloud.

3.3. Score and Prediction Function Module

The fused feature y 1 f u n s i o n is processed through an outer fully connected layer and is then subjected to a SoftMax operation to derive the scores for each point.
The prediction module is primarily composed of a classification module and a regression module, as illustrated in Figure 6. The classification module comprises three distinct MLPs, each responsible for estimating different poses of the human head. These MLPs function independently, with the parameter “ λ ” determining the number of neurons in each MLP. On the other hand, the regression module computes the final head pose prediction by summing the product of the representative values for each classification interval based on the classification results.
The prediction module initially learns data distribution by utilizing a classification layer and discretizing the regression space. Head pose angles are then directly converted into data distribution and anticipated values within predetermined classification intervals. These predicted values undergo weighting with corresponding angles to compute the head pose angles for regression loss. Subsequently, the contribution of various data distribution values to network training are adjusted by employing a weighted combination of classification and regression. Following the output production by the SoftMax layer, KL divergence is used to assess the similarity between the true and predicted label distributions. Additionally, to quantify disparities between the actual and estimated head poses, we employed expected regression. The proposed head pose estimation loss function is defined as follows:
L = k y k l n y ^ k + α × n = 1 k p k × y ^ k y
where y represents the true head pose angle values; y ^ k represents the distribution predicted by the network; p k denotes the representative values of the angle intervals corresponding to each category in the network’s predicted distribution labels, and α serves as an essential weight parameter that balances the two components of the loss function.

4. Experimental Results

4.1. Data Source

This study utilized the BIWI dataset [35], which includes over 15,000 RGB-D images at a resolution of 640 × 480. The dataset contained information from more than twenty individuals (six females and fourteen males) with head yaw angles starting from ±75° and pitch angles from ±60°. Each frame within the dataset consists of an RGB image, a depth image, corresponding ground-truth head pose parameters, and three-dimensional positional parameters for the human head center. The BIWI dataset is commonly used for training 3D head pose models.

4.2. Experimental Parameters

The experimental setup comprised an 11th Gen Intel(R) Core(TM) i5-1135G7 2.40 GHz processor, 16.0 GB of system RAM, and a 64-bit Windows 10 operating system. It also included a GTX 3060 GPU with 12 GB of VRAM, and the development environment was based on Python 3.7, using the Torch 1.71 framework. The initial step in this study involved using a face detector to identify facial regions, followed by cropping the image based on the detected face’s position. Subsequently, the network employed the cropped RGB head image and the depth image to convert the human head into point cloud data and extract point cloud features. The interval angle λ was initially set to 1° when creating head pose labels, resulting in 200 classes with β set to 3. Discrete distribution labels were computed for each image in the BIWI dataset. The loss function, which combined classification and regression components, was used to update network parameters via backpropagation. The weight coefficient α in the loss function was initially set to 0.5. Experiments were conducted using the first sixteen datasets from the BIWI dataset as the training set, with the remaining eight datasets utilized as test set α . Cross-validation was performed using sets 1, 7, 13, and 19 as individual test sets to underscore the reliability of the proposed method. Additionally, for a comparative analysis with the prevalent head pose estimation methods, datasets 11 and 12 were used as the test sets, while the remaining data served as the training set. Notably, no data augmentation was applied in these experiments owing to the nature of the point cloud data generated from the depth images, representing only the surface of the human body, which would alter with image rotation. The selected hyperparameters included a batch size of 16, an initial learning rate of 0.001, and a decay in the learning rate of 0.7 every 40 epochs. The training process spanned 100 epochs, using the mean absolute error defined in Equation (8) as the evaluation metric:
M A E = 1 N n = 1 N Y p r e d i c t Y r e a l
where N represents the total number of frames; Y p r e d i c t represents the head pose estimated by the model, and Y r e a l represents the true head pose angles.

4.3. Experimental Results

4.3.1. Data Processing Result

We utilized the head mask and head center position information from the BIWI dataset to derive candidate boxes for the head, thereby generating point cloud data, as depicted in Figure 7.
The volume of point cloud data significantly impacts the accuracy and speed of head pose estimation. Point clouds containing an excessively large quantity of data often encompass redundant information. Directly inputting such voluminous data into the network can lead to prolonged computation times. Conversely, having too few points in the point cloud may result in the loss of local feature information. To strike a balance between accuracy and enhanced model inference speed, this study uniformly downsampled the point cloud data to 512 points, as demonstrated in Figure 8.

4.3.2. Ablation Experiment Results

This study explored the values of α and conducted experiments with α sets of 0.1, 1, and 2. The initial 16 sets from the BIWI dataset constituted the training set, while the remaining sets served as the test sets for the experiments. Table 1 showcases the experimental results.
The α value represents the weight of the expected regression loss within the overall loss function. When this weight is low, the network tends to focus more on retaining the distribution of discrete data, potentially affecting accurate predictions. Conversely, when the weight is high, the network may learn the data themselves but could fail to capture the relationships between adjacent elements of the true labels α . Setting α to 0.1 resulted in the model exhibiting its poorest performance on the test set, as indicated in Table 1. In contrast, setting α to 1 led to the highest accuracy achieved. Therefore, for the subsequent experiments in this study, α was set to 1. Regarding the self-attention mechanisms, a comparison was drawn between training a standalone PointNet network and a PointNet network. The results of this comparison are presented in Table 2.
In the table, “S-A” denotes the incorporation of spatial attention mechanisms, while “R” and “P” represent the data sources as images and point clouds, respectively. The results indicate that integrating self-attention mechanisms into PointNet enhanced the accuracy of head pose estimation. Moreover, the fusion of point cloud features with image features for head pose estimation yielded a higher accuracy compared to estimating the head pose from a single data source. Additionally, encoding the position of the network input enabled the model to capture more high-frequency information, thereby enhancing the accuracy of head pose estimation. To verify the effectiveness of this approach, comparative experiments were conducted, using the 1st, 3rd, 6th, 8th, 17th, and 23rd groups as test sets. Simultaneously, the remaining data comprised the training set. Figure 9 presents the experimental results.
Evidently, head pose estimation utilizing positional encoding in the model’s input showcased more accurate results compared to head pose estimation without positional encoding. The most notable enhancement was observed when employing the 6th group of data as the test set, resulting in an accuracy increase of approximately 3° for both the roll and yaw angles.

4.3.3. Model Evaluation and Comparison

To authenticate the reliability of the proposed method, it was juxtaposed with other mainstream methods for head pose estimation. The experiments utilized the 11th and 12th groups as test sets, while the remaining groups constituted the training sets. Table 3 depicts the experimental results.
As observed, the accuracy of head pose estimation using depth images surpassed that of RGB images due to the richer information present in-depth images. Moreover, head pose estimation utilizing point cloud data achieved even higher accuracy than the preceding methods. This is primarily attributed to the more pronounced spatial information within the point cloud data compared to the depth images. Among the nine head pose estimation methods scrutinized, the approach proposed in this study yielded the most accurate estimation results. Figure 10 juxtaposes the ground-truth values with the predicted values of the yaw, roll, and pitch angles within the test set. Notably, when handling both small and large pose angles, the head poses estimated by the method outlined in this study closely corresponded to the ground-truth values. Using the Gumbel distribution instead of the Gaussian distribution led to an increase of 0.27° in the pitch angle and 0.06° in the roll angle.
To evaluate the model’s generalizability, each of the 24 data groups was individually used as a test set, while the remaining groups served for training, as illustrated in Figure 11. Across the 24 test experiments, it became apparent that the average absolute errors for each angle were all below 3°. Notably, the 7th, 11th, 12th, and 13th groups showcased average fundamental errors of less than 1° for each angle. These experimental outcomes strongly suggested that the proposed method surpassed the BIWI head pose recognition accuracy.
In [18], the authors attained the most favorable head pose estimation outcomes by integrating PointNet++ and GCNN. For the comparative analysis, cross-validation was conducted, designating each dataset as a test set. In this study, similar cross-validation techniques were employed, using each dataset as a test set, and the obtained experimental outcomes were compared with those detailed in [23]. The comparative results are illustrated in Figure 12.
For head pose estimation, point cloud data contain more information than RGB data, while the global information extracted from image features can complement the point cloud information. Experimental results also confirmed that the method presented in this study consistently achieved higher head pose estimation accuracy than the approach described in [23], with errors for head pose angles in groups 1, 13, and 19 all being less than 1°.
This study provides visual results of the head pose estimation for small and large pose angles in the 11th and 12th groups to demonstrate the approach’s advantages more intuitively. Figure 13 shows that the proposed method can provide accurate predictions even in significant facial occlusion conditions and large pose angles. This reflects the robustness of the approach for facial occlusion conditions and large head pose angles.

5. Discussion

5.1. Comparison with Existing Methods

In the above experimental results, this paper compared its method with those from references [15,16,23,36,37,38,39] on the BIWI dataset, and the experimental results are shown in Table 3 From the results, it can be observed that the model proposed in this paper achieved a higher accuracy. From Table 3, it is evident that the accuracy of estimating head pose solely from RGB images is lower compared to the accuracy obtained from depth images. This is because depth images contain more information compared to RGB images. Furthermore, head pose estimation using point cloud data achieved an even higher accuracy than the previous two methods. This is primarily because of the more pronounced spatial information in the point cloud data than in the depth images.
Additionally, this paper conducted cross-validation on the BIWI dataset and independently compared the proposed method with the model in reference [23]. Experimental results demonstrate that the approach presented in this paper performs optimally across all test sets. Moreover, even in cases where large pose angles lead to the absence of head point clouds, the proposed method effectively estimates head pose. This suggests that image features can serve as a complement to missing point cloud information. Furthermore, favorable head pose estimation results can be achieved through a judicious feature fusion method.

5.2. Future Research

The point cloud and image features extracted by different networks are distinct. Rich feature information is beneficial for head pose estimation accuracy; however, redundant features can decrease the inference speed of the model. This study utilized improved PointNet networks for point-cloud feature extraction and employed ResNet networks to extract image features. This paper did not introduce any improvements in the image feature extraction network. In feature fusion, the consideration of time complexity was also omitted. Using image features as a supplement to the missing information in the point cloud can enhance the accuracy of head pose estimation. Still, it comes at the cost of increased model inference time.
In the future, researchers should explore models more suitable to extracting features from point clouds and images. Additionally, concerning the fusion of these features, consideration will be given to time complexity and feature redundancy. One approach involves early mapping of the point cloud onto the image, followed by extracting features from the corresponding pixel positions of the point cloud in later stages for fusion.

6. Conclusions

We generated head point clouds from depth maps and discretized head pose labels using the Gumbel function. The MLP in PointNet was substituted with a residual MLP module, incorporating a self-attention mechanism to extract point cloud features. Additionally, we utilized a ResNet network to extract global features from images. The features from both sources were then fused, and the SoftMax function was applied to compute scores for each point, identifying the features of the point with the highest score. During prediction, the model input the highest score features into a classification block and estimated the final pose through a fully connected layer using a softmax function. The experimental validation of the proposed method on the BIWI dataset resulted in errors of 1.18°, 0.67°, and 0.68° for the yaw, pitch, and roll angles, respectively. Compared to widely used head pose estimation methods, the proposed method demonstrated a significantly improved accuracy in head pose estimation.

Author Contributions

Conceptualization, K.C. and Z.W.; methodology, K.C. and Z.W.; software, J.H.; validation, J.H.; formal analysis, K.C. and J.H.; investigation, K.C. and Y.S.; resources, Z.W. and J.H.; data curation, K.C.; writing—original draft preparation, Z.W. and J.H.; writing—review and editing, K.C. and Y.S.; visualization, Z.W. and J.H.; supervision, K.C. and J.H.; project administration, Z.W. and J.H.; funding acquisition, Z.W. and J.H. All authors have read and agreed to the published version of the manuscript.

Funding

This work was partially supported by the “Open Fund of Key Laboratory of Geospatial Technology for the Middle and Lower Yellow River Regions (Henan University), Ministry of Education under Grant” [GTYR202209] and the “University-Enterprise Collaboration Project(Hefei University of Technology)” [W2023JSFW0173].

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Informed consent was obtained from all the subjects involved in the study.

Data Availability Statement

The data presented in this study are openly available in Random Forests for Real Time Head Pose Estimation (ethz.ch).

Conflicts of Interest

The authors declare no conflict of interest. The sponsors had no role in the design, execution, interpretation, or writing of this study.

References

  1. Rossi, S.; Leone, E.; Staffa, M. Using random forests for the estimation of multiple users’ visual focus of attention from head pose. In Proceedings of the XV of AI* IA 2016 Advances in Artificial Intelligence: XVth International Conference of the Italian Association for Artificial Intelligence, Genova, Italy, 29 November–1 December 2016. [Google Scholar]
  2. Huang, S.; Yang, K.; Xiao, H.; Han, P.; Qiu, J.; Peng, L.; Liu, D.; Luo, K. A new head pose tracking method based on stereo visual SLAM. J. Vis. Commun. Image Represent. 2022, 82, 103402. [Google Scholar] [CrossRef]
  3. Liu, H.; Liu, T.; Zhang, Z.; Sangaiah, A.K.; Yang, B.; Li, Y. ARHPE: Asymmetric Relation-Aware Representation Learning for Head Pose Estimation in Industrial Human-Computer Interaction. IEEE Trans. Ind. Inf. 2022, 18, 7107–7117. [Google Scholar] [CrossRef]
  4. Avola, D.; Cinque, L.; Del Bimbo, A.; Marini, M.R. MIFTel: A multimodal interactive framework based on temporal logic rules. Multimed. Tools Appl. 2020, 79, 13533–13558. [Google Scholar] [CrossRef]
  5. Liu, H.; Nie, H.; Zhang, Z.; Li, Y.-F. Anisotropic angle distribution learning for head pose estimation and attention understanding in human-computer interaction. Neurocomputing 2021, 433, 310–322. [Google Scholar] [CrossRef]
  6. Wongphanngam, J.; Pumrin, S. Fatigue warning system for driver nodding off using depth image from Kinect. In Proceedings of the 2016 13th International Conference on Electrical Engineering/Electronics, Computer, Telecommunications and Information Technology, Chiang Mai, Thailand, 28 June–1 July 2016; pp. 1–6. [Google Scholar]
  7. Baltrušaitis, T.; Robinson, P.; Morency, L.P. OpenFace: An open source facial behavior analysis toolkit. In Proceedings of the 2016 IEEE Winter Conference on Applications of Computer Vision, Lake Placid, NY, USA, 7–10 March 2016; pp. 1–10. [Google Scholar]
  8. Han, J.; Luo, K.; Qiu, J.; Liu, D.; Peng, L.; Han, P. Head attitude estimation method of eye tracker based on binocular camera. Adv. Laser Optoelectron. 2019, 58, 310–317. [Google Scholar]
  9. Zhao, G.; Chen, L.; Song, J.; Chen, G. Large head movement tracking using sift-based registration. In Proceedings of the 15th ACM International Conference on Multimedia, Augsburg, Germany, 25–29 September 2007; pp. 807–810. [Google Scholar]
  10. Liu, L.; Ke, Z.; Huo, J.; Chen, J. Head pose estimation through keypoints matching between reconstructed 3D face model and 2D image. Sensors 2021, 21, 1841. [Google Scholar] [CrossRef]
  11. Liu, H.; Zhang, C.; Deng, Y.; Liu, T.; Zhang, Z.; Li, Y.F. Orientation Cues-Aware Facial Relationship Representation for Head Pose Estimation via Transformer. IEEE Trans. Image Process. 2023, 32, 6289–6302. [Google Scholar] [CrossRef]
  12. Geng, X.; Qian, X.; Huo, Z.; Zhang, Y. Head pose estimation based on multivariate label distribution. IEEE Trans. Pattern Anal. Mach. Intell. 2020, 44, 1974–1991. [Google Scholar] [CrossRef]
  13. Zhang, C.; Liu, H.; Deng, Y.; Xie, B.; Li, Y. TokenHPE: Learning Orientation Tokens for Efficient Head Pose Estimation via Transformers. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada, 18–22 June 2023; pp. 8897–8906. [Google Scholar]
  14. Liu, H.; Fang, S.; Zhang, Z.; Li, D.; Lin, K.; Wang, J. MFDNet: Collaborative Poses Perception and Matrix Fisher Distribution for Head Pose Estimation. IEEE Trans. Multimedia 2022, 24, 2449–2460. [Google Scholar] [CrossRef]
  15. Ruiz, N.; Chong, E.; Rehg, J.M. Fine-grained head pose estimation without keypoints. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, Salt Lake City, UT, USA, 18–22 June 2018; pp. 2074–2083. [Google Scholar]
  16. Yang, T.-Y.; Chen, Y.-T.; Lin, Y.-Y.; Chuang, Y.-Y. FSA-Net: Learning fine-grained structure aggregation for head pose estimation from a single image. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Long Beach, CA, USA, 15–20 June 2019; pp. 1087–1096. [Google Scholar]
  17. Zhang, H.; Zhang, Y.; Geng, X. Practical age estimation using deep label distribution learning. Front. Comput. Sci. 2021, 15, 153318. [Google Scholar] [CrossRef]
  18. Liu, T.; Wang, J.; Yang, B.; Wang, X. NGDNet: Nonuniform Gaussian-label distribution learning for infrared head pose estimation and on-task behavior understanding in the classroom. Neurocomputing 2021, 436, 210–220. [Google Scholar] [CrossRef]
  19. Xu, L.; Chen, J.; Gan, Y. Head pose estimation with soft labels using regularized convolutional neural network. Neurocomputing 2017, 337, 339–353. [Google Scholar] [CrossRef]
  20. Chenglong, L.; Fan, Z.; Xin, M.; Xeuying, Q. Real-time head attitude estimation based on Kalman filter and random regression forest. J. Comput. Aid. Des. Graph. 2017, 29, 2309–2316. [Google Scholar]
  21. Wang, Y.; Yuan, G.; Fu, X. Driver’s head pose and gaze zone estimation based on multi-zone templates registration and multi-frame point cloud fusion. Sensors 2022, 22, 3154. [Google Scholar] [CrossRef]
  22. Shihua, X.; Nan, S.; Xupeng, W. 3D point cloud head attitude estimation based on Deep learning. J. Comput. Appl. 2020, 40, 996–1001. (In Chinese) [Google Scholar]
  23. Xu, Y.; Jung, C.; Chang, Y. Head pose estimation using deep neural networks and 3D point clouds. Pattern Recog. 2022, 121, 108210. [Google Scholar] [CrossRef]
  24. Zhang, Y.; Fu, K.; Wang, J.; Cheng, P. Learning from discrete Gaussian label distribution and spatial channel-aware residual attention for head pose estimation. Neurocomputing 2020, 407, 259–269. [Google Scholar] [CrossRef]
  25. Gumbel, E.J. Les valeurs extrêmes des distributions statistiques. Ann. De L’Institut Henri Poincaré 1935, 5, 115–158. [Google Scholar]
  26. Hornik, K.; Stinchcombe, M.; White, H. Multilayer feedforward networks are universal approximators. Neural Netw. 1989, 2, 359–366. [Google Scholar] [CrossRef]
  27. Mildenhall, B.; Srinivasan, P.P.; Tancik, M.; Barron, J.T.; Ramamoorthi, R.; Ng, R. NeRF: Representing scenes as neural radiance fields for view synthesis. Commun. ACM 2021, 65, 99–106. [Google Scholar] [CrossRef]
  28. Charles, R.Q.; Su, H.; Mo, K.; Guibas, L.J. PointNet: Deep learning on point sets for 3D classification and segmentation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA, 21–26 July 2017; pp. 652–660. [Google Scholar]
  29. He, K.; Zhang, X.; Ren, S.; Sun, J. Deep residual learning for image recognition. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, 27–30 June 2016; pp. 770–778. [Google Scholar]
  30. Chevtchenko, S.F.; Vale, R.F.; Macario, V.; Cordeiro, F.R. A convolutional neural network with feature fusion for real-time hand posture recognition. Appl. Soft Comput. 2018, 73, 748–766. [Google Scholar] [CrossRef]
  31. Zhou, W.; Dong, S.; Lei, J.; Yu, L. MTANet: Multitask-aware network with hierarchical multimodal fusion for RGB-T urban scene understanding. IEEE Trans. Intell. Veh. 2022, 8, 48–58. [Google Scholar] [CrossRef]
  32. Xu, D.; Anguelov, D.; Jain, A. PointFusion: Deep sensor fusion for 3D bounding box estimation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–23 June 2018. [Google Scholar]
  33. Wang, C.; Xu, D.; Zhu, Y.; Martín-Martín, R.; Lu, C.; Fei-Fei, L.; Savarese, S. DenseFusion: 6D object pose estimation by iterative dense fusion. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern recognition, Long Beach, CA, USA, 15–20 June 2019. [Google Scholar]
  34. Liu, H.; Wang, X.; Zhang, W.; Zhang, Z.; Li, Y.-F. Infrared head pose estimation with multi-scales feature fusion on the IRHP database for human attention recognition. Neurocomputing 2020, 411, 510–520. [Google Scholar] [CrossRef]
  35. Fanelli, G.; Gall, J.; Gool, L.V. Real time head pose estimation with random regression forests. In Proceedings of the Conference on Computer Vision and Pattern Recognition 2011, Colorado Springs, CO, USA, 20–25 June 2011. [Google Scholar]
  36. Xu, X.; Kakadiaris, I.A. Joint head pose estimation and face alignment framework using global and local CNN features. In Proceedings of the 2017 12th IEEE International Conference on Automatic Face & Gesture Recognition (FG 2017), Washington, DC, USA, 30 May–3 June 2017. [Google Scholar]
  37. Wang, Y.; Liang, W.; Shen, J.; Jia, Y.; Yu, L.-F. A deep coarse-to-fine network for head pose estimation from synthetic data. Pattern Recog. 2019, 94, 196–206. [Google Scholar] [CrossRef]
  38. Borghi, G.; Fabbri, M.; Vezzani, R.; Calderara, S.; Cucchiara, R. Face-from-depth for head pose estimation on depth images. IEEE Trans. Pattern Anal. Mach. Intell. 2018, 42, 596–609. [Google Scholar] [CrossRef]
  39. Meyer, G.P.; Gupta, S.; Frosio, I.; Reddy, D.; Kautz, J. Robust model-based 3D head pose estimation. In Proceedings of the IEEE International Conference on Computer Vision, Santiago, Chile, 7–13 December 2015. [Google Scholar]
Figure 1. The top is the RGB image, and the bottom is the head pose label corresponding to the RGB image. (a) When the yaw angle is 38.27°, the discretized results of the head pose. (b) When the yaw angle is −40.39°, the discretized results of the head pose.
Figure 1. The top is the RGB image, and the bottom is the head pose label corresponding to the RGB image. (a) When the yaw angle is 38.27°, the discretized results of the head pose. (b) When the yaw angle is −40.39°, the discretized results of the head pose.
Sensors 23 09894 g001
Figure 2. Head pose estimation network. Different colors represent the characteristics of different stages. The network is mainly divided into four modules: feature function, fusion function, score function, and predict function.
Figure 2. Head pose estimation network. Different colors represent the characteristics of different stages. The network is mainly divided into four modules: feature function, fusion function, score function, and predict function.
Sensors 23 09894 g002
Figure 3. Feedforward residual MLP module.
Figure 3. Feedforward residual MLP module.
Sensors 23 09894 g003
Figure 4. Spatial self-attention module.
Figure 4. Spatial self-attention module.
Sensors 23 09894 g004
Figure 5. Feature extraction module. The network consists of five residual point blocks, three attention blocks, and a feature transform.
Figure 5. Feature extraction module. The network consists of five residual point blocks, three attention blocks, and a feature transform.
Sensors 23 09894 g005
Figure 6. Classification and regression module.
Figure 6. Classification and regression module.
Sensors 23 09894 g006
Figure 7. Data source. (a) RGB image. The head position in the RGB image. (b) Head mask. In the head mask, the white region is the head, while the black region is the background. (c) Point cloud. From the depth map to the point cloud.
Figure 7. Data source. (a) RGB image. The head position in the RGB image. (b) Head mask. In the head mask, the white region is the head, while the black region is the background. (c) Point cloud. From the depth map to the point cloud.
Sensors 23 09894 g007
Figure 8. Point clouds at different scales. (a) Primary point cloud. (b) Downsampling the original point cloud to 1024. (c) Downsampling the original point cloud to 512.
Figure 8. Point clouds at different scales. (a) Primary point cloud. (b) Downsampling the original point cloud to 1024. (c) Downsampling the original point cloud to 512.
Sensors 23 09894 g008
Figure 9. When comparing model prediction accuracy, within the same dataset, between instances with and without positional encoding. (a) The predictive accuracy of the model on yaw angles. (b) The predictive accuracy of the model on pitch angles. (c) The predictive accuracy of the model on roll angles.
Figure 9. When comparing model prediction accuracy, within the same dataset, between instances with and without positional encoding. (a) The predictive accuracy of the model on yaw angles. (b) The predictive accuracy of the model on pitch angles. (c) The predictive accuracy of the model on roll angles.
Sensors 23 09894 g009
Figure 10. When using the 11th and 12th instances as the dataset, a comparison between the model’s predicted values and the ground truth. (a) Comparison on yaw angles. (b) Comparison on pitch angles. (c) Comparison on roll angles.
Figure 10. When using the 11th and 12th instances as the dataset, a comparison between the model’s predicted values and the ground truth. (a) Comparison on yaw angles. (b) Comparison on pitch angles. (c) Comparison on roll angles.
Sensors 23 09894 g010
Figure 11. When each set of data in the dataset is used as a test set, the model’s predicted results are compared to the ground truth, and the average absolute error is calculated. The red dotted line represents the upper limit of the model’s prediction accuracy.
Figure 11. When each set of data in the dataset is used as a test set, the model’s predicted results are compared to the ground truth, and the average absolute error is calculated. The red dotted line represents the upper limit of the model’s prediction accuracy.
Sensors 23 09894 g011
Figure 12. Comparison of different methods on the BIWI dataset. (a) The comparison of different methods in terms of yaw angle prediction accuracy. (b) The comparison of different methods in terms of pitch angle prediction accuracy. (c) The comparison of different methods in terms of roll angle prediction accuracy.
Figure 12. Comparison of different methods on the BIWI dataset. (a) The comparison of different methods in terms of yaw angle prediction accuracy. (b) The comparison of different methods in terms of pitch angle prediction accuracy. (c) The comparison of different methods in terms of roll angle prediction accuracy.
Sensors 23 09894 g012
Figure 13. Visualization of partial test set results. The blue, green, and red colors respectively represent yaw, pitch, and roll angles. The top and bottom sections show the RGB and point cloud visualizations, respectively.
Figure 13. Visualization of partial test set results. The blue, green, and red colors respectively represent yaw, pitch, and roll angles. The top and bottom sections show the RGB and point cloud visualizations, respectively.
Sensors 23 09894 g013
Table 1. Influence of α on head pose estimation.
Table 1. Influence of α on head pose estimation.
α Yaw (°)Pitch (°)Roll (°)Mean (°)
0.101.671.852.261.92
1.001.381.341.761.49
2.001.361.431.851.54
Table 2. Comparison of head pose estimations obtained for different data sources and methods.
Table 2. Comparison of head pose estimations obtained for different data sources and methods.
MethodYaw (°)Pitch (°)Roll (°)Mean (°)
S-A + P + R1.381.341.761.49
S-A + P1.622.052.071.91
P1.442.592.172.06
Table 3. Comparison between different head pose estimation methods. PGCNN stands for point graph convolutional neural networks.
Table 3. Comparison between different head pose estimation methods. PGCNN stands for point graph convolutional neural networks.
DataYaw (°)Pitch (°)Roll (°)Mean (°)
FineGrained [15]RGB3.293.393.303.23
FSANet [16]RGB4.964.272.764.00
PGCNN [23]Point1.821.091.391.42
Multi-task [36]RGB4.303.603.403.76
CoarseFine [37]RGB4.765.484.294.84
POSEidon [38]Depth1.701.601.801.70
RobustMode [39]Depth2.402.202.102.10
Our (Gauss)RGB + Point1.120.940.740.93
Our (Gumbel)RGB + Point1.180.670.680.84
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Chen, K.; Wu, Z.; Huang, J.; Su, Y. Self-Attention Mechanism-Based Head Pose Estimation Network with Fusion of Point Cloud and Image Features. Sensors 2023, 23, 9894. https://doi.org/10.3390/s23249894

AMA Style

Chen K, Wu Z, Huang J, Su Y. Self-Attention Mechanism-Based Head Pose Estimation Network with Fusion of Point Cloud and Image Features. Sensors. 2023; 23(24):9894. https://doi.org/10.3390/s23249894

Chicago/Turabian Style

Chen, Kui, Zhaofu Wu, Jianwei Huang, and Yiming Su. 2023. "Self-Attention Mechanism-Based Head Pose Estimation Network with Fusion of Point Cloud and Image Features" Sensors 23, no. 24: 9894. https://doi.org/10.3390/s23249894

APA Style

Chen, K., Wu, Z., Huang, J., & Su, Y. (2023). Self-Attention Mechanism-Based Head Pose Estimation Network with Fusion of Point Cloud and Image Features. Sensors, 23(24), 9894. https://doi.org/10.3390/s23249894

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop