adetailer

Bingsu ultralytics

Bingsu/adetailer

15,501,796

下载量

884

收藏数

18

浏览量

apache-2.0

许可

简介

- coco2017（仅人物） - AniSeg - skytnt/anime-segmentation

模型卡片

许可协议 apache-2.0

框架 ultralytics

数据集

wider_face anime-segmentation

pytorch

模型详情

已翻译

YOLOv8 检测模型

数据集

人脸

手部

人体

deepfashion2

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

信息

模型	目标	mAP 50	mAP 50-95
face_yolov8n.pt	2D / 真实人脸	0.660	0.366
face_yolov8n_v2.pt	2D / 真实人脸	0.669	0.372
face_yolov8s.pt	2D / 真实人脸	0.713	0.404
face_yolov8m.pt	2D / 真实人脸	0.737	0.424
face_yolov9c.pt	2D / 真实人脸	0.748	0.433
hand_yolov8n.pt	2D / 真实手部	0.767	0.505
hand_yolov8s.pt	2D / 真实手部	0.794	0.527
hand_yolov9c.pt	2D / 真实手部	0.810	0.550
person_yolov8n-seg.pt	2D / 真实人体	0.782 (bbox)0.761 (mask)	0.555 (bbox)0.460 (mask)
person_yolov8s-seg.pt	2D / 真实人体	0.824 (bbox)0.809 (mask)	0.605 (bbox)0.508 (mask)
person_yolov8m-seg.pt	2D / 真实人体	0.849 (bbox)0.831 (mask)	0.636 (bbox)0.533 (mask)
deepfashion2_yolov8s-seg.pt	真实服装	0.849 (bbox)0.840 (mask)	0.763 (bbox)0.675 (mask)

使用方法

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

不安全文件

由于 getattr 被归类为危险的 pickle 函数，任何使用它的分割模型都被归类为不安全。

所有模型均使用官方 ultralytics 库创建和保存，因此使用从可信来源下载的文件是安全的。

另请参阅：https://huggingface.co/docs/hub/security-pickle