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"), } )