Upload 11 files
Browse files- README.md +88 -1
- deepfashion2_yolov8s-seg.pt +3 -0
- face_yolov8m.pt +3 -0
- face_yolov8n.pt +3 -0
- face_yolov8n_v2.pt +3 -0
- face_yolov8s.pt +3 -0
- hand_yolov8n.pt +3 -0
- hand_yolov8s.pt +3 -0
- person_yolov8m-seg.pt +3 -0
- person_yolov8n-seg.pt +3 -0
- person_yolov8s-seg.pt +3 -0
README.md
CHANGED
@@ -1,3 +1,90 @@
|
|
1 |
---
|
2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|
deepfashion2_yolov8s-seg.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c039144a13fb7ae5306eb11846c97ba6dac53337a52a6a459c16d42f9ba5f917
|
3 |
+
size 23819714
|
face_yolov8m.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e3893a92c5c1907136b6cc75404094db767c1e0cfefe1b43e87dad72af2e4c9f
|
3 |
+
size 51996128
|
face_yolov8n.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:30e7adcf03c1e250ffce66a9ff5324358c311f02a093f93f5424d650803ce077
|
3 |
+
size 6207800
|
face_yolov8n_v2.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1e561e2edfcc99d815c8489d3770fef52b8d46325de70d0a96f3950ec05a9023
|
3 |
+
size 6215608
|
face_yolov8s.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2e157766211db78c724f69f8b289c9edcf5a2c0a3956298a51fb4b94debd2658
|
3 |
+
size 22484600
|
hand_yolov8n.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7f2af8cacc1064769f98abe309bc6eadf900e86cf5b2a27260ec8dea65443eeb
|
3 |
+
size 6215544
|
hand_yolov8s.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:30878cea9870964d4a238339e9dcff002078bbbaa1a058b07e11c167f67eca1c
|
3 |
+
size 22484536
|
person_yolov8m-seg.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1fd7e562f240a5debd48bf737753de6fb60c63f8664121bb522f090a885d8254
|
3 |
+
size 54791722
|
person_yolov8n-seg.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9bb5af28e2f583d97318f67ce67341807073a8ac6a72d4f1efad877cab6656f5
|
3 |
+
size 6747842
|
person_yolov8s-seg.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:67104fdd78933192d2613cfc2bd0842a3d2b75076d258d97d751381b3e0019cf
|
3 |
+
size 23820738
|