Capability enum change COLOR=>TONE.
This commit is contained in:
@@ -12,7 +12,7 @@ class Capability(StrEnum):
|
||||
AGE = "age"
|
||||
AGEGROUP = "age_group"
|
||||
SEX = "sex"
|
||||
SKINCOLOR = "skin_color"
|
||||
SKINTONE = "skin_tone"
|
||||
ETHNICITY = "ethinicity"
|
||||
|
||||
|
||||
@@ -23,7 +23,9 @@ class InvalidCapabilityError(Exception):
|
||||
|
||||
class BaseEstimator(ABC):
|
||||
@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.
|
||||
|
||||
The input `images` is a dictionary indexed by the image name, or ID,
|
||||
|
||||
Reference in New Issue
Block a user