Commit
·
3dfc475
1
Parent(s):
ff04b51
Update README.md
Browse files
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 |
+

|