JCTN
/

PyTorch
ultralytics
JCTN commited on
Commit
9adfa47
1 Parent(s): 2ba86fa

Upload 11 files

Browse files
README.md CHANGED
@@ -1,3 +1,90 @@
1
  ---
2
- license: other
 
 
 
 
 
 
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