Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-29 18:51:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: missing mediapipe confidences

Browse Source
This commit is contained in:
Dowon
2024-11-13 19:48:31 +09:00
parent 8ddf919e1e
commit 891fb9469d
3 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_mediapipe.py
View File

@@ -16,3 +16,6 @@ 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

3
tests/test_ultralytics.py
View File

@@ -48,3 +48,6 @@ def test_yolo_world(sample_image2: Image.Image, klass: str):
model_path = hf_hub_download("Bingsu/yolo-world-mirror", "yolov8x-worldv2.pt")
result = ultralytics_predict(model_path, sample_image2, classes=klass)
assert result.preview is not None
assert len(result.bboxes) > 0
assert len(result.masks) > 0
assert len(result.confidences) > 0