Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-01-26 11:19:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix tests

Browse Source 
mediapipe only failed on macos
This commit is contained in:
Dowon
2024-12-02 22:35:35 +09:00
parent cb69a52fb3
commit 6a0e501654
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_mediapipe.py
View File

@@ -15,7 +15,8 @@ from adetailer.mediapipe import mediapipe_predict
)
def test_mediapipe(sample_image2: Image.Image, model_name: str):
result = mediapipe_predict(model_name, sample_image2)
assert result.preview is not None
assert len(result.bboxes) > 0
assert len(result.masks) > 0
assert len(result.confidences) > 0
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)

9
tests/test_ultralytics.py
View File

@@ -25,12 +25,20 @@ def test_ultralytics_hf_models(sample_image: Image.Image, model_name: str):
model_path = hf_hub_download("Bingsu/adetailer", model_name)
result = ultralytics_predict(model_path, sample_image)
assert 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)
def test_yolo_world_default(sample_image: Image.Image):
model_path = hf_hub_download("Bingsu/yolo-world-mirror", "yolov8x-worldv2.pt")
result = ultralytics_predict(model_path, sample_image)
assert 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)
@pytest.mark.parametrize(
@@ -51,3 +59,4 @@ def test_yolo_world(sample_image2: Image.Image, klass: str):
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)