UniFace: A Comprehensive Library for Face Detection, Recognition, Tracking, Landmark Analysis, Face Parsing, Gaze Estimation, Age, and Gender Detection
Project description
UniFace: All-in-One Face Analysis Library
UniFace is a lightweight, production-ready face analysis library built on ONNX Runtime. It provides high-performance face detection, recognition, landmark detection, face parsing, gaze estimation, and attribute analysis with hardware acceleration support across platforms.
Features
- Face Detection — RetinaFace, SCRFD, YOLOv5-Face, and YOLOv8-Face with 5-point landmarks
- Face Recognition — ArcFace, MobileFace, and SphereFace embeddings
- Face Tracking — Multi-object tracking with BYTETracker for persistent IDs across video frames
- Facial Landmarks — 106-point landmark localization module (separate from 5-point detector landmarks)
- Face Parsing — BiSeNet semantic segmentation (19 classes), XSeg face masking
- Gaze Estimation — Real-time gaze direction with MobileGaze
- Head Pose Estimation — 3D head orientation (pitch, yaw, roll) with 6D rotation representation
- Attribute Analysis — Age, gender, race (FairFace), and emotion
- Vector Indexing — FAISS-backed embedding store for fast multi-identity search
- Anti-Spoofing — Face liveness detection with MiniFASNet
- Face Anonymization — 5 blur methods for privacy protection
- Hardware Acceleration — ARM64 (Apple Silicon), CUDA (NVIDIA), CPU
Installation
Standard installation
pip install uniface
GPU support (CUDA)
pip install uniface[gpu]
From source (latest version)
git clone https://github.com/yakhyo/uniface.git
cd uniface && pip install -e .
FAISS vector indexing
pip install faiss-cpu # or faiss-gpu for CUDA
Optional dependencies
- Emotion model uses TorchScript and requires
torch:pip install torch(choose the correct build for your OS/CUDA) - YOLOv5-Face and YOLOv8-Face support faster NMS with
torchvision:pip install torch torchvisionthen usenms_mode='torchvision'
Model Downloads and Cache
Models are downloaded automatically on first use and verified via SHA-256.
Default cache location: ~/.uniface/models
Override with the programmatic API or environment variable:
from uniface.model_store import get_cache_dir, set_cache_dir
set_cache_dir('/data/models')
print(get_cache_dir()) # /data/models
export UNIFACE_CACHE_DIR=/data/models
Quick Example (Detection)
import cv2
from uniface.detection import RetinaFace
detector = RetinaFace()
image = cv2.imread("photo.jpg")
if image is None:
raise ValueError("Failed to load image. Check the path to 'photo.jpg'.")
faces = detector.detect(image)
for face in faces:
print(f"Confidence: {face.confidence:.2f}")
print(f"BBox: {face.bbox}")
print(f"Landmarks: {face.landmarks.shape}")
Face Detection Model Output
Example (Face Analyzer)
import cv2
from uniface.analyzer import FaceAnalyzer
from uniface.detection import RetinaFace
from uniface.recognition import ArcFace
detector = RetinaFace()
recognizer = ArcFace()
analyzer = FaceAnalyzer(detector, recognizer=recognizer)
image = cv2.imread("photo.jpg")
if image is None:
raise ValueError("Failed to load image. Check the path to 'photo.jpg'.")
faces = analyzer.analyze(image)
for face in faces:
print(face.bbox, face.embedding.shape if face.embedding is not None else None)
Execution Providers (ONNX Runtime)
from uniface.detection import RetinaFace
# Force CPU-only inference
detector = RetinaFace(providers=["CPUExecutionProvider"])
See more in the docs: https://yakhyo.github.io/uniface/concepts/execution-providers/
Documentation
Full documentation: https://yakhyo.github.io/uniface/
| Resource | Description |
|---|---|
| Quickstart | Get up and running in 5 minutes |
| Model Zoo | All models, benchmarks, and selection guide |
| API Reference | Detailed module documentation |
| Tutorials | Step-by-step workflow examples |
| Guides | Architecture and design principles |
| Datasets | Training data and evaluation benchmarks |
Datasets
| Task | Training Dataset | Models |
|---|---|---|
| Detection | WIDER FACE | RetinaFace, SCRFD, YOLOv5-Face, YOLOv8-Face |
| Recognition | MS1MV2 | MobileFace, SphereFace |
| Recognition | WebFace600K | ArcFace |
| Recognition | WebFace4M / 12M | AdaFace |
| Gaze | Gaze360 | MobileGaze |
| Head Pose | 300W-LP | HeadPose (ResNet, MobileNet) |
| Parsing | CelebAMask-HQ | BiSeNet |
| Attributes | CelebA, FairFace, AffectNet | AgeGender, FairFace, Emotion |
See Datasets documentation for download links, benchmarks, and details.
Jupyter Notebooks
| Example | Colab | Description |
|---|---|---|
| 01_face_detection.ipynb | Face detection and landmarks | |
| 02_face_alignment.ipynb | Face alignment for recognition | |
| 03_face_verification.ipynb | Compare faces for identity | |
| 04_face_search.ipynb | Find a person in group photos | |
| 05_face_analyzer.ipynb | All-in-one analysis | |
| 06_face_parsing.ipynb | Semantic face segmentation | |
| 07_face_anonymization.ipynb | Privacy-preserving blur | |
| 08_gaze_estimation.ipynb | Gaze direction estimation | |
| 09_face_segmentation.ipynb | Face segmentation with XSeg | |
| 10_face_vector_store.ipynb | FAISS-backed face database | |
| 11_head_pose_estimation.ipynb | Head pose estimation (pitch, yaw, roll) |
Licensing and Model Usage
UniFace is MIT-licensed, but several pretrained models carry their own licenses. Review: https://yakhyo.github.io/uniface/license-attribution/
Notable examples:
- YOLOv5-Face and YOLOv8-Face weights are GPL-3.0
- FairFace weights are CC BY 4.0
If you plan commercial use, verify model license compatibility.
References
| Feature | Repository | Training | Description |
|---|---|---|---|
| Detection | retinaface-pytorch | ✓ | RetinaFace PyTorch Training & Export |
| Detection | yolov5-face-onnx-inference | - | YOLOv5-Face ONNX Inference |
| Detection | yolov8-face-onnx-inference | - | YOLOv8-Face ONNX Inference |
| Tracking | bytetrack-tracker | - | BYTETracker Multi-Object Tracking |
| Recognition | face-recognition | ✓ | MobileFace, SphereFace Training |
| Parsing | face-parsing | ✓ | BiSeNet Face Parsing |
| Parsing | face-segmentation | - | XSeg Face Segmentation |
| Gaze | gaze-estimation | ✓ | MobileGaze Training |
| Head Pose | head-pose-estimation | ✓ | Head Pose Training (6DRepNet-style) |
| Anti-Spoofing | face-anti-spoofing | - | MiniFASNet Inference |
| Attributes | fairface-onnx | - | FairFace ONNX Inference |
*SCRFD and ArcFace models are from InsightFace.
Contributing
Contributions are welcome. Please see CONTRIBUTING.md.
Support
If you find this project useful, consider giving it a ⭐ on GitHub — it helps others discover it!
Questions or feedback:
- Discord: https://discord.gg/wdzrjr7R5j
- GitHub Issues: https://github.com/yakhyo/uniface/issues
- DeepWiki Q&A: https://deepwiki.com/yakhyo/uniface
License
This project is licensed under the MIT License.
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
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file uniface-3.2.0.tar.gz.
File metadata
- Download URL: uniface-3.2.0.tar.gz
- Upload date:
- Size: 88.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a27143a09ee081f861b7c31fdfd4a8c14470afb9de5a82fa0073b879b90f1e71
|
|
| MD5 |
e578e52d1ab10eaf2e0807b8a88d54b0
|
|
| BLAKE2b-256 |
fa5694978e4fdee122ec7dd26e82e5376f212f14990c649c378ade27720727a0
|
File details
Details for the file uniface-3.2.0-py3-none-any.whl.
File metadata
- Download URL: uniface-3.2.0-py3-none-any.whl
- Upload date:
- Size: 109.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.11.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7ca9abf1e11f87e49848cd37880f0495e3427a3a28d62fe76605c46437b3ef33
|
|
| MD5 |
86811b30870aa26e975beb1c1c3fb121
|
|
| BLAKE2b-256 |
8a0b50c8b4ae2355d0c9aab0781eda2dc41053c2df2a6f6646ee2ec2efe850e5
|