feat: Add Face Parsing model BiSeNet model trained on CelebMask dataset (#35)

* Add BiSeNet face parsing implementation

* Add parsing model weights configuration

* Export BiSeNet in main package

* Add face parsing tests

* Add face parsing examples and script

* Bump version to 1.5.0

* Update documentation for face parsing

* Fix face parsing notebook to use lips instead of mouth

* chore: Update the face parsing example

* fix: Fix model argument to use Enum

* ref: Move vis_parsing_map function into visualization.py

* docs: Update README.md
This commit is contained in:
Yakhyokhuja Valikhujaev
2025-12-14 21:13:53 +09:00
committed by GitHub
parent 4d1921e531
commit 54b769c0f1
13 changed files with 1267 additions and 7 deletions

View File

@@ -1,7 +1,7 @@
[project]
name = "uniface"
version = "1.4.0"
description = "UniFace: A Comprehensive Library for Face Detection, Recognition, Landmark Analysis, Gaze Estimation, Age, and Gender Detection"
version = "1.5.0"
description = "UniFace: A Comprehensive Library for Face Detection, Recognition, Landmark Analysis, Face Parsing, Gaze Estimation, Age, and Gender Detection"
readme = "README.md"
license = { text = "MIT" }
authors = [{ name = "Yakhyokhuja Valikhujaev", email = "yakhyo9696@gmail.com" }]
@@ -14,6 +14,8 @@ keywords = [
"face-detection",
"face-recognition",
"facial-landmarks",
"face-parsing",
"face-segmentation",
"gaze-estimation",
"age-detection",
"gender-detection",
@@ -22,6 +24,7 @@ keywords = [
"onnx",
"onnxruntime",
"face-analysis",
"bisenet",
]
classifiers = [