ultralytics
PyTorch
kylielee505 commited on
Commit
e57eb83
·
verified ·
1 Parent(s): 49a1190

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -25,7 +25,6 @@
25
  *.safetensors filter=lfs diff=lfs merge=lfs -text
26
  saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
  *.tar.* filter=lfs diff=lfs merge=lfs -text
28
- *.tar filter=lfs diff=lfs merge=lfs -text
29
  *.tflite filter=lfs diff=lfs merge=lfs -text
30
  *.tgz filter=lfs diff=lfs merge=lfs -text
31
  *.wasm filter=lfs diff=lfs merge=lfs -text
 
25
  *.safetensors filter=lfs diff=lfs merge=lfs -text
26
  saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
  *.tar.* filter=lfs diff=lfs merge=lfs -text
 
28
  *.tflite filter=lfs diff=lfs merge=lfs -text
29
  *.tgz filter=lfs diff=lfs merge=lfs -text
30
  *.wasm filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.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
+ | face_yolov9c.pt | 2D / realistic face | 0.748 | 0.433 |
62
+ | hand_yolov8n.pt | 2D / realistic hand | 0.767 | 0.505 |
63
+ | hand_yolov8s.pt | 2D / realistic hand | 0.794 | 0.527 |
64
+ | hand_yolov9c.pt | 2D / realistic hand | 0.810 | 0.550 |
65
+ | person_yolov8n-seg.pt | 2D / realistic person | 0.782 (bbox)<br/>0.761 (mask) | 0.555 (bbox)<br/>0.460 (mask) |
66
+ | person_yolov8s-seg.pt | 2D / realistic person | 0.824 (bbox)<br/>0.809 (mask) | 0.605 (bbox)<br/>0.508 (mask) |
67
+ | person_yolov8m-seg.pt | 2D / realistic person | 0.849 (bbox)<br/>0.831 (mask) | 0.636 (bbox)<br/>0.533 (mask) |
68
+ | deepfashion2_yolov8s-seg.pt | realistic clothes | 0.849 (bbox)<br/>0.840 (mask) | 0.763 (bbox)<br/>0.675 (mask) |
69
+
70
+ ## Usage
71
+
72
+ ```python
73
+ from huggingface_hub import hf_hub_download
74
+ from ultralytics import YOLO
75
+
76
+ path = hf_hub_download("Bingsu/adetailer", "face_yolov8n.pt")
77
+ model = YOLO(path)
78
+ ```
79
+
80
+ ```python
81
+ import cv2
82
+ from PIL import Image
83
+
84
+ img = "https://farm5.staticflickr.com/4139/4887614566_6b57ec4422_z.jpg"
85
+ output = model(img)
86
+ pred = output[0].plot()
87
+ pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
88
+ pred = Image.fromarray(pred)
89
+ pred
90
+ ```
91
+
92
+ ![image](https://i.imgur.com/9ny1wmD.png)
93
+
94
+
95
+ ## Unsafe files
96
+
97
+ ![image](https://i.imgur.com/9Btuy8j.png)
98
+
99
+ Since `getattr` is classified as a dangerous pickle function, any segmentation model that uses it is classified as unsafe.
100
+
101
+ All models were created and saved using the official [ultralytics](https://github.com/ultralytics/ultralytics) library, so it's okay to use files downloaded from a trusted source.
102
+
103
+ See also: https://huggingface.co/docs/hub/security-pickle
deepfashion2_yolov8s-seg.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c319c15e86443ea61f70b40620909d737c47ff2a39381536503ac5c54772f8ac
3
+ size 23852321
face_yolov8m.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:717923c19b3f4bbf5250b728f1fa6b2cb72a33aed1d236ea9caf0e21ad943e5f
3
+ size 52026019
face_yolov8n.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70b640f8f60b1cf0dcc72f30caf3da9495eb2fb6509da48c53374ad6806e6a9c
3
+ size 6230011
face_yolov8n_v2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f5f2110f83c4e00712993fab48c771d26036e2e80ec62bd5b9cb37c29e36b36
3
+ size 6239036
face_yolov8s.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7237eff25787377de196961140ceaed324d859ee8de5a775d93d33a0e3fab78
3
+ size 22507707
face_yolov9c.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d02fe493c31e1bbc6450f4dc6f1db86a02a59322ff1f6d318da0661d72ddd084
3
+ size 51648019
hand_yolov8n.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3991202eb69e9ddcb3b9ba80cdeb41e734ffaf844403d6c9f47d515cd88c6f29
3
+ size 6237883
hand_yolov8s.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70b540063fbc385736d8258970744a4afbc4cbf7932134bae3b24cdadeadec06
3
+ size 22507643
hand_yolov9c.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f116f686ef5942fa0c1d08db481b3d7622047a97847eccfb2a264f9ce9777f2
3
+ size 51633747
person_yolov8m-seg.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8ab26f517173b1fe8342d336a09f443eb61cb08dcbfc78d53fff4c2547ae81e
3
+ size 54827683
person_yolov8n-seg.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38fc8aaae97cb6e70be4ec44770005b26ed473471362afcda62a0037d7ccf432
3
+ size 6777003
person_yolov8s-seg.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53c54aec2239355faffc6c5b70d0f3d05042f386f956cbec39cec46ad456f050
3
+ size 23850731