adetailer
Bingsu's ultralytics-based face and body detection model collection used for automated detail refinement in image generation pipelines.
Model Card
Datasets
Face
Hand
Person
- coco2017 (only person)
- AniSeg
- skytnt/anime-segmentation
deepfashion2
| id | label |
|---|---|
| 0 | short_sleeved_shirt |
| 1 | long_sleeved_shirt |
| 2 | short_sleeved_outwear |
| 3 | long_sleeved_outwear |
| 4 | vest |
| 5 | sling |
| 6 | shorts |
| 7 | trousers |
| 8 | skirt |
| 9 | short_sleeved_dress |
| 10 | long_sleeved_dress |
| 11 | vest_dress |
| 12 | sling_dress |
Info
| Model | Target | mAP 50 | mAP 50-95 |
|---|---|---|---|
| face_yolov8n.pt | 2D / realistic face | 0.660 | 0.366 |
| face_yolov8n_v2.pt | 2D / realistic face | 0.669 | 0.372 |
| face_yolov8s.pt | 2D / realistic face | 0.713 | 0.404 |
| face_yolov8m.pt | 2D / realistic face | 0.737 | 0.424 |
| face_yolov9c.pt | 2D / realistic face | 0.748 | 0.433 |
| hand_yolov8n.pt | 2D / realistic hand | 0.767 | 0.505 |
| hand_yolov8s.pt | 2D / realistic hand | 0.794 | 0.527 |
| hand_yolov9c.pt | 2D / realistic hand | 0.810 | 0.550 |
| 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) | |||
| person_yolov8m-seg.pt | 2D / realistic person | 0.849 (bbox) | |
| 0.831 (mask) | 0.636 (bbox) | ||
| 0.533 (mask) | |||
| deepfashion2_yolov8s-seg.pt | realistic clothes | 0.849 (bbox) | |
| 0.840 (mask) | 0.763 (bbox) | ||
| 0.675 (mask) |
Usage
from huggingface_hub import hf_hub_download
from ultralytics import YOLO
path = hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt")
model = YOLO(path)
import cv2
from PIL import Image
img = "https://farm5.staticflickr.com/4139/4887614566_6b57ec4422_z.jpg"
output = model(img)
pred = output[0].plot()
pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
pred = Image.fromarray(pred)
pred
Unsafe files
Since getattr is classified as a dangerous pickle function, any segmentation model that uses it is classified as unsafe.
All models were created and saved using the official ultralytics library, so it's okay to use files downloaded from a trusted source.
Author
Get the full context.
Sign up to read complete case studies, access detailed metrics, and unlock all use cases.
Author
Get the full context.
Sign up to read complete case studies, access detailed metrics, and unlock all use cases.