Dhivimhotep commited on
Commit
6c89bf8
1 Parent(s): d8816e4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -90
README.md CHANGED
@@ -1,90 +0,0 @@
1
- ---
2
- license: agpl-3.0
3
- library_name: ultralytics
4
- datasets:
5
- - wider_face
6
- - skytnt/anime-segmentation
7
- tags:
8
- - pytorch
9
- ---
10
-
11
- # YOLOv8 Detection Model
12
-
13
- ## Datasets
14
-
15
- ### Face
16
-
17
- - [Anime Face CreateML](https://universe.roboflow.com/my-workspace-mph8o/anime-face-createml)
18
- - [xml2txt](https://universe.roboflow.com/0oooooo0/xml2txt-njqx1)
19
- - [AN](https://universe.roboflow.com/sed-b8vkf/an-lfg5i)
20
- - [wider face](http://shuoyang1213.me/WIDERFACE/index.html)
21
-
22
- ### Hand
23
-
24
- - [AnHDet](https://universe.roboflow.com/1-yshhi/anhdet)
25
- - [hand-detection-fuao9](https://universe.roboflow.com/catwithawand/hand-detection-fuao9)
26
-
27
- ### Person
28
-
29
- - [coco2017](https://cocodataset.org/#home) (only person)
30
- - [AniSeg](https://github.com/jerryli27/AniSeg)
31
- - [skytnt/anime-segmentation](https://huggingface.co/datasets/skytnt/anime-segmentation)
32
-
33
- ### deepfashion2
34
-
35
- - [deepfashion2](https://github.com/switchablenorms/DeepFashion2)
36
-
37
- | id | label |
38
- | --- | --------------------- |
39
- | 0 | short_sleeved_shirt |
40
- | 1 | long_sleeved_shirt |
41
- | 2 | short_sleeved_outwear |
42
- | 3 | long_sleeved_outwear |
43
- | 4 | vest |
44
- | 5 | sling |
45
- | 6 | shorts |
46
- | 7 | trousers |
47
- | 8 | skirt |
48
- | 9 | short_sleeved_dress |
49
- | 10 | long_sleeved_dress |
50
- | 11 | vest_dress |
51
- | 12 | sling_dress |
52
-
53
- ## Info
54
-
55
- | Model | Target | mAP 50 | mAP 50-95 |
56
- | --------------------------- | --------------------- | ----------------------------- | ----------------------------- |
57
- | face_yolov8n.pt | 2D / realistic face | 0.660 | 0.366 |
58
- | face_yolov8n_v2.pt | 2D / realistic face | 0.669 | 0.372 |
59
- | face_yolov8s.pt | 2D / realistic face | 0.713 | 0.404 |
60
- | face_yolov8m.pt | 2D / realistic face | 0.737 | 0.424 |
61
- | hand_yolov8n.pt | 2D / realistic hand | 0.767 | 0.505 |
62
- | hand_yolov8s.pt | 2D / realistic hand | 0.794 | 0.527 |
63
- | person_yolov8n-seg.pt | 2D / realistic person | 0.782 (bbox)<br/>0.761 (mask) | 0.555 (bbox)<br/>0.460 (mask) |
64
- | person_yolov8s-seg.pt | 2D / realistic person | 0.824 (bbox)<br/>0.809 (mask) | 0.605 (bbox)<br/>0.508 (mask) |
65
- | person_yolov8m-seg.pt | 2D / realistic person | 0.849 (bbox)<br/>0.831 (mask) | 0.636 (bbox)<br/>0.533 (mask) |
66
- | deepfashion2_yolov8s-seg.pt | realistic clothes | 0.849 (bbox)<br/>0.840 (mask) | 0.763 (bbox)<br/>0.675 (mask) |
67
-
68
- ## Usage
69
-
70
- ```python
71
- from huggingface_hub import hf_hub_download
72
- from ultralytics import YOLO
73
-
74
- path = hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt")
75
- model = YOLO(path)
76
- ```
77
-
78
- ```python
79
- import cv2
80
- from PIL import Image
81
-
82
- img = "https://farm5.staticflickr.com/4139/4887614566_6b57ec4422_z.jpg"
83
- output = model(img)
84
- pred = output[0].plot()
85
- pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
86
- pred = Image.fromarray(pred)
87
- pred
88
- ```
89
-
90
- ![image](https://i.imgur.com/9ny1wmD.png)