File size: 5,724 Bytes
8a0af76 401a153 7a3dcec 401a153 55595b1 7a3dcec 5029ed6 8a0af76 7a1ee24 3f53c64 3ebf5bd 4e98433 6c66e9f 3ebf5bd 4e98433 3f53c64 3ebf5bd 3f53c64 3ebf5bd 6c66e9f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
---
license: agpl-3.0
tags:
- pytorch
- YOLOv8
- art
- Ultralytics
base_model:
- Ultralytics/YOLOv8
library_name: ultralytics
pipeline_tag: object-detection
---
## Available Models
### Face segmentation:
#### Universal:
Series of models aiming at detecting and segmenting face accurately. Trained on closed dataset i annotated myself.
| Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
| --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
| Anzhc Face -seg.pt | Face: illustration, real | LOST DATA | LOST DATA |2(male, female)|LOST DATA| 640|
| Anzhc Face -seg-hd.pt | Face: illustration, real | LOST DATA | LOST DATA |2(male, female)|LOST DATA|1024|
| Anzhc Face seg 640 v2 y8n.pt | Face: illustration, real | 0.872(box),0.872(mask) | 0.835(box),0.752(mask)|1(face) |~500| 640|
| Anzhc Face seg 768 v2 y8n.pt | Face: illustration, real | 0.86(box),0.86(mask) | 0.81(box),0.726(mask) |1(face) |~500| 768|
| Anzhc Face seg 768MS v2 y8n.pt | Face: illustration, real | 0.866(box),0.866(mask) | 0.816(box),0.72(mask) |1(face) |~500| 768|(Multi-scale)|
| Anzhc Face seg 1024 v2 y8n.pt | Face: illustration, real | 0.872(box),0.872(mask) | 0.804(box),0.726(mask)|1(face) |~500| 1024|
Take those stats with a grain of salt, since im pretty sure i re-scrambled dataset partition after training those models ages ago.
Benchmark was performed in 640px.
Difference in v2 models are only in their target resolution, so their performance spread is marginal.
#### Real Face, gendered:
Trained only on real photos for the most part, so will perform poorly with illustrations, but is gendered, and can be used for male/female detection stack.
| Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
| --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
| Anzhcs ManFace v02 1024 y8n.pt | Face: real | 0.883(box),0.883(mask) | 0.778(box), 0.704(mask) |1(face) |~340 |1024|
| Anzhcs WomanFace v05 1024 y8n.pt | Face: real | 0.82(box),0.82(mask) | 0.713(box), 0.659(mask) |1(face) |~600 |1024|
Benchmark was performed in 640px.
### Eyes segmnetation:
PLACEHOLDER
| Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
| --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
| Anzhcs ManFace v02 1024 y8n.pt | Face: real | 0.883(box),0.883(mask) | 0.778(box), 0.704(mask) |1(face) |~340 |1024|
| Anzhcs WomanFace v05 1024 y8n.pt | Face: real | 0.82(box),0.82(mask) | 0.713(box), 0.659(mask) |1(face) |~600 |1024|
### Head+Hair segmentation:
PLACEHOLDER
| Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
| --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
| Anzhcs ManFace v02 1024 y8n.pt | Face: real | 0.883(box),0.883(mask) | 0.778(box), 0.704(mask) |1(face) |~340 |1024|
| Anzhcs WomanFace v05 1024 y8n.pt | Face: real | 0.82(box),0.82(mask) | 0.713(box), 0.659(mask) |1(face) |~600 |1024|
### Breasts segmentation:
Model for segmenting breasts. Was trained on anime images only, therefore has very weak realistic performance, but still is possible.
| Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
| --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
| Anzhc Breasts Seg v1 1024n.pt | Breasts: illustration | 0.742(box),0.73(mask) | 0.563(box), 0.535(mask) |1(breasts) |~2000 |1024|
| Anzhc Breasts Seg v1 1024s.pt | Breasts: illustration | 0.768(box),0.763(mask) | 0.596(box), 0.575(mask) |1(breasts) |~2000 |1024|
| Anzhc Breasts Seg v1 1024m.pt | Breasts: illustration | 0.782(box),0.775(mask) | 0.644(box), 0.614(mask) |1(breasts) |~2000 |1024|
/--UNDER CONSTRUCTION--/ |