Commit
·
7aa0a61
1
Parent(s):
f50a7fa
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 |
+

|
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
|