mirror of
https://github.com/Bing-su/adetailer.git
synced 2026-02-11 02:40:04 +00:00
19 lines
444 B
Python
19 lines
444 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)
|
|
assert result.preview is not None
|