Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-03-11 08:20:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add person_yolov8s-seg.pt model

Browse Source
This commit is contained in:
Bingsu
2023-05-09 16:59:45 +09:00
parent 4b3e5f7e96
commit 52ce9a721f
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@@ -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)<br/>0.761 (mask) | 0.555 (bbox)<br/>0.460 (mask) |
| person_yolov8s-seg.pt | 2D / realistic person | 0.824 (bbox)<br/>0.809 (mask) | 0.605 (bbox)<br/>0.508 (mask) |
The yolo models can be found on huggingface [Bingsu/adetailer](https://huggingface.co/Bingsu/adetailer).

1
adetailer/common.py
View File

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