mirror of
https://github.com/Bing-su/adetailer.git
synced 2026-01-26 19:29:54 +00:00
23 lines
641 B
Python
23 lines
641 B
Python
import pytest
|
|
from PIL import Image
|
|
|
|
from adetailer.mediapipe import mediapipe_predict
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"model_name",
|
|
[
|
|
"mediapipe_face_short",
|
|
"mediapipe_face_full",
|
|
"mediapipe_face_mesh",
|
|
"mediapipe_face_mesh_eyes_only",
|
|
],
|
|
)
|
|
def test_mediapipe(sample_image2: Image.Image, model_name: str):
|
|
result = mediapipe_predict(model_name, sample_image2)
|
|
if result.preview is not None:
|
|
assert len(result.bboxes) > 0
|
|
assert len(result.masks) > 0
|
|
assert len(result.confidences) > 0
|
|
assert len(result.bboxes) == len(result.masks) == len(result.confidences)
|