Bengal-HP_RU: A Dataset of Bengal People For Head Pose Estimation
Description
1. Accurate head pose estimation depends critically on the availability of large-scale, diverse, and representative training data. While several head pose datasets exist in the literature, they predominantly feature subjects of Western or East Asian origin, with little to no representation of South Asian populations, particularly Bengali individuals. Facial geometry, skin tone distribution, and cultural appearance attributes such as attire and accessories vary across ethnic groups, and models trained on non-representative datasets often generalise poorly to underrepresented demographics. 2. To address this, Bengal-HP RU is introduced as the first publicly available head pose dataset centred on Bengali subjects, offering in-the-wild image conditions with rich diversity in age, gender, occlusion patterns, and illumination, making it a valuable resource for training and evaluating robust head pose estimation models in real-world scenarios. 3. Images are annotated with continuous yaw, pitch, and roll values through a two-stage process combining automated geometric estimation with manual correction. 4. A total of 3,591 source images were collected from Wikimedia Commons using Bengali-relevant keyword searches and processed through an automated pipeline using YOLOv8-Face for face detection and MediaPipe for landmark extraction, yielding 12,894 cropped head images partitioned into 10,494 training and 2,400 test images across 296 unique uploaders. 5. The dataset is organised into train and test subsets, each containing cropped head images and corresponding per-image .npz files storing the face mesh, 68 facial landmarks, face bounding box, and computed Euler angles, alongside train.csv and test.csv files providing source image metadata. 6. The full directory structure is as follows: Bengal-HP_RU/ ├── train.csv ├── test.csv ├── demo.py # Example script for loading NPZ annotations and visualizing them on an image ├── train/ │ ├── images_train/ # 10,494 cropped head images │ └── npz_train/ # 10,494 per-image .npz files └── test/ ├── images_test/ # 2,400 cropped head images └── npz_test/ # 2,400 per-image .npz files
Files
Institutions
- University of RajshahiRajshahi Division, Rajshahi