From 52ce9a721f60e9a653bcd42875bdc09d53686f61 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Tue, 9 May 2023 16:59:45 +0900 Subject: [PATCH] feat: add person_yolov8s-seg.pt model --- README.md | 1 + adetailer/common.py | 1 + 2 files changed, 2 insertions(+) diff --git a/README.md b/README.md index 3b40d8a..36117f7 100644 --- a/README.md +++ b/README.md @@ -36,6 +36,7 @@ On the ControlNet tab, select a ControlNet inpaint model and set the model weigh | mediapipe_face_short | realistic face | - | - | | hand_yolov8n.pt | 2D / realistic hand | 0.767 | 0.505 | | person_yolov8n-seg.pt | 2D / realistic person | 0.782 (bbox)
0.761 (mask) | 0.555 (bbox)
0.460 (mask) | +| person_yolov8s-seg.pt | 2D / realistic person | 0.824 (bbox)
0.809 (mask) | 0.605 (bbox)
0.508 (mask) | The yolo models can be found on huggingface [Bingsu/adetailer](https://huggingface.co/Bingsu/adetailer). diff --git a/adetailer/common.py b/adetailer/common.py index fe91b3f..5b03da5 100644 --- a/adetailer/common.py +++ b/adetailer/common.py @@ -39,6 +39,7 @@ def get_models(model_dir: Union[str, Path]) -> OrderedDict[str, Optional[str]]: "mediapipe_face_short": None, "hand_yolov8n.pt": hf_hub_download(repo_id, "hand_yolov8n.pt"), "person_yolov8n-seg.pt": hf_hub_download(repo_id, "person_yolov8n-seg.pt"), + "person_yolov8s-seg.pt": hf_hub_download(repo_id, "person_yolov8s-seg.pt"), } )