guilherme.schardong/facebias
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Capability enum change COLOR=>TONE.

Browse Source
This commit is contained in:
Guilherme Gonçalves Schardong
2026-05-06 16:28:28 +01:00
parent e2790d2d5d
commit 536c29978d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/facebias/estimators/__init__.py
View File

@@ -12,7 +12,7 @@ class Capability(StrEnum):
AGE = "age" AGE = "age"
AGEGROUP = "age_group" AGEGROUP = "age_group"
SEX = "sex" SEX = "sex"
SKINCOLOR = "skin_color" SKINTONE = "skin_tone"
ETHNICITY = "ethinicity" ETHNICITY = "ethinicity"
@@ -23,7 +23,9 @@ class InvalidCapabilityError(Exception):
class BaseEstimator(ABC): class BaseEstimator(ABC):
@abstractmethod @abstractmethod
def predict(self, images: dict[Path, np.ndarray]) -> dict[Path, dict[Capability, Any]]: def predict(
self, images: dict[Path, np.ndarray]
) -> dict[Path, dict[Capability, Any]]:
"""Runs the estimator on a batch of images. """Runs the estimator on a batch of images.
The input `images` is a dictionary indexed by the image name, or ID, The input `images` is a dictionary indexed by the image name, or ID,