From ea29140cdf1020409393f4b4a92d4e1ea73a7280 Mon Sep 17 00:00:00 2001 From: Dowon Date: Sat, 16 Mar 2024 21:23:41 +0900 Subject: [PATCH] feat: update yolo world to v2 --- adetailer/common.py | 4 ++-- tests/test_ultralytics.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/adetailer/common.py b/adetailer/common.py index 9e1d1b9..5fc6b53 100644 --- a/adetailer/common.py +++ b/adetailer/common.py @@ -57,8 +57,8 @@ def get_models( "hand_yolov8n.pt": hf_download("hand_yolov8n.pt"), "person_yolov8n-seg.pt": hf_download("person_yolov8n-seg.pt"), "person_yolov8s-seg.pt": hf_download("person_yolov8s-seg.pt"), - "yolov8x-world.pt": hf_download( - "yolov8x-world.pt", repo_id="Bingsu/yolo-world-mirror" + "yolov8x-worldv2.pt": hf_download( + "yolov8x-worldv2.pt", repo_id="Bingsu/yolo-world-mirror" ), } ) diff --git a/tests/test_ultralytics.py b/tests/test_ultralytics.py index fe40097..1a75d7a 100644 --- a/tests/test_ultralytics.py +++ b/tests/test_ultralytics.py @@ -26,7 +26,7 @@ def test_ultralytics_hf_models(sample_image: Image.Image, model_name: str): def test_yolo_world_default(sample_image: Image.Image): - model_path = hf_hub_download("Bingsu/yolo-world-mirror", "yolov8x-world.pt") + 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 @@ -43,6 +43,6 @@ def test_yolo_world_default(sample_image: Image.Image): ], ) def test_yolo_world(sample_image2: Image.Image, klass: str): - model_path = hf_hub_download("Bingsu/yolo-world-mirror", "yolov8x-world.pt") + 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