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

feat: add yolo world model

Browse Source
This commit is contained in:
Dowon
2024-03-01 10:54:19 +09:00
parent 692b052cfb
commit 9e9dcd5bca
7 changed files with 38 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_ultralytics.py
View File

@@ -28,7 +28,7 @@ def test_ultralytics_hf_models(sample_image: Image.Image, model_name: str):
def test_yolo_world_default(sample_image: Image.Image):
result = ultralytics_predict("yolov8l-world.pt", sample_image)
result = ultralytics_predict("yolov8x-world.pt", sample_image)
assert result.preview is not None
@@ -44,5 +44,5 @@ def test_yolo_world_default(sample_image: Image.Image):
],
)
def test_yolo_world(sample_image2: Image.Image, klass: str):
result = ultralytics_predict("yolov8l-world.pt", sample_image2, classes=klass)
result = ultralytics_predict("yolov8x-world.pt", sample_image2, classes=klass)
assert result.preview is not None