Head pose estimation module
Project description
State of art the Head Pose Estimation in Tensorflow2
This repository includes:
-
"WHENet: Real-time Fine-Grained Estimation for Wide Range Head Pose" (BMVC 2020). adapted from the original source code.
-
RetinaFace: Single-stage Dense Face Localisation in the Wild adapted from https://github.com/StanislasBertrand/RetinaFace-tf2.
Install
You can install this repository with pip (requires python>=3.6);
pip install headpose_estimation
pip install git+https://github.com/geekysethi/headpose_estimation
You can also install with the setup.py
With Face Detection
To perform detection you can simple use the following lines:
import cv2
from headpose_estimation import Headpose
headpose = Headpose()
img = cv2.imread("path_to_im.jpg")
detections,image = headpose.run(img)
This will return a list of dictionary which looks like this [{'bbox': [xmin, ymin, xmax, ymax], 'yaw': yaw_value, 'pitch': pitch_value, 'roll': roll_value}
Without Face Detection
To perform detection you can simple use the following lines:
import cv2
from headpose_estimation import Headpose
headpose = Headpose(face_detection=False)
imgcrop = cv2.imread("path_to_im.jpg")
detections,image = headpose.run(imgcrop)
In this case it will return a list of dictionary which looks like this [{'yaw': yaw_value, 'pitch': pitch_value, 'roll': roll_value}
Dependncies
- EfficientNet https://github.com/qubvel/efficientnet
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for headpose_detection-0.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 647e9ad232c380419665179d6f3ca1668d949b3f33ba7ff260a2a53a802fc9b6 |
|
MD5 | b550e9f044ca9d69d0f21612ea4d15ac |
|
BLAKE2b-256 | 31aa2b359cb464ec91305d496d33a830fd453aeb1f74a5393fa0312d093d42c6 |