mirror of
https://github.com/yakhyo/uniface.git
synced 2025-12-30 09:02:25 +00:00
0c93598007
* feat: Update linting and type annotations, return types in detect * feat: add face analyzer and face classes * chore: Update the format and clean up some docstrings * docs: Update usage documentation * feat: Change AgeGender model output to 0, 1 instead of string (Female, Male) * test: Update testing code * feat: Add Apple silicon backend for torchscript inference * feat: Add face analyzer example and add run emotion for testing
118 lines
4.1 KiB
Python
118 lines
4.1 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
from uniface.attribute import AgeGender
|
|
|
|
|
|
@pytest.fixture
|
|
def age_gender_model():
|
|
return AgeGender()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_image():
|
|
return np.random.randint(0, 255, (640, 640, 3), dtype=np.uint8)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_bbox():
|
|
return [100, 100, 300, 300]
|
|
|
|
|
|
def test_model_initialization(age_gender_model):
|
|
assert age_gender_model is not None, 'AgeGender model initialization failed.'
|
|
|
|
|
|
def test_prediction_output_format(age_gender_model, mock_image, mock_bbox):
|
|
gender_id, age = age_gender_model.predict(mock_image, mock_bbox)
|
|
assert isinstance(gender_id, int), f'Gender ID should be int, got {type(gender_id)}'
|
|
assert isinstance(age, int), f'Age should be int, got {type(age)}'
|
|
|
|
|
|
def test_gender_values(age_gender_model, mock_image, mock_bbox):
|
|
gender_id, age = age_gender_model.predict(mock_image, mock_bbox)
|
|
assert gender_id in [0, 1], f'Gender ID should be 0 (Female) or 1 (Male), got {gender_id}'
|
|
|
|
|
|
def test_age_range(age_gender_model, mock_image, mock_bbox):
|
|
gender_id, age = age_gender_model.predict(mock_image, mock_bbox)
|
|
assert 0 <= age <= 120, f'Age should be between 0 and 120, got {age}'
|
|
|
|
|
|
def test_different_bbox_sizes(age_gender_model, mock_image):
|
|
test_bboxes = [
|
|
[50, 50, 150, 150],
|
|
[100, 100, 300, 300],
|
|
[50, 50, 400, 400],
|
|
]
|
|
|
|
for bbox in test_bboxes:
|
|
gender_id, age = age_gender_model.predict(mock_image, bbox)
|
|
assert gender_id in [0, 1], f'Failed for bbox {bbox}'
|
|
assert 0 <= age <= 120, f'Age out of range for bbox {bbox}'
|
|
|
|
|
|
def test_different_image_sizes(age_gender_model, mock_bbox):
|
|
test_sizes = [(480, 640, 3), (720, 1280, 3), (1080, 1920, 3)]
|
|
|
|
for size in test_sizes:
|
|
mock_image = np.random.randint(0, 255, size, dtype=np.uint8)
|
|
gender_id, age = age_gender_model.predict(mock_image, mock_bbox)
|
|
assert gender_id in [0, 1], f'Failed for image size {size}'
|
|
assert 0 <= age <= 120, f'Age out of range for image size {size}'
|
|
|
|
|
|
def test_consistency(age_gender_model, mock_image, mock_bbox):
|
|
gender_id1, age1 = age_gender_model.predict(mock_image, mock_bbox)
|
|
gender_id2, age2 = age_gender_model.predict(mock_image, mock_bbox)
|
|
|
|
assert gender_id1 == gender_id2, 'Same input should produce same gender prediction'
|
|
assert age1 == age2, 'Same input should produce same age prediction'
|
|
|
|
|
|
def test_bbox_list_format(age_gender_model, mock_image):
|
|
bbox_list = [100, 100, 300, 300]
|
|
gender_id, age = age_gender_model.predict(mock_image, bbox_list)
|
|
assert gender_id in [0, 1], 'Should work with bbox as list'
|
|
assert 0 <= age <= 120, 'Age should be in valid range'
|
|
|
|
|
|
def test_bbox_array_format(age_gender_model, mock_image):
|
|
bbox_array = np.array([100, 100, 300, 300])
|
|
gender_id, age = age_gender_model.predict(mock_image, bbox_array)
|
|
assert gender_id in [0, 1], 'Should work with bbox as numpy array'
|
|
assert 0 <= age <= 120, 'Age should be in valid range'
|
|
|
|
|
|
def test_multiple_predictions(age_gender_model, mock_image):
|
|
bboxes = [
|
|
[50, 50, 150, 150],
|
|
[200, 200, 350, 350],
|
|
[400, 400, 550, 550],
|
|
]
|
|
|
|
results = []
|
|
for bbox in bboxes:
|
|
gender_id, age = age_gender_model.predict(mock_image, bbox)
|
|
results.append((gender_id, age))
|
|
|
|
assert len(results) == 3, 'Should have 3 predictions'
|
|
for gender_id, age in results:
|
|
assert gender_id in [0, 1]
|
|
assert 0 <= age <= 120
|
|
|
|
|
|
def test_age_is_positive(age_gender_model, mock_image, mock_bbox):
|
|
for _ in range(5):
|
|
gender_id, age = age_gender_model.predict(mock_image, mock_bbox)
|
|
assert age >= 0, f'Age should be non-negative, got {age}'
|
|
|
|
|
|
def test_output_format_for_visualization(age_gender_model, mock_image, mock_bbox):
|
|
gender_id, age = age_gender_model.predict(mock_image, mock_bbox)
|
|
gender_str = 'Female' if gender_id == 0 else 'Male'
|
|
text = f'{gender_str}, {age}y'
|
|
assert isinstance(text, str), 'Should be able to format as string'
|
|
assert 'Male' in text or 'Female' in text, 'Text should contain gender'
|
|
assert 'y' in text, "Text should contain 'y' for years"
|