Anzhc commited on
Commit
d22e2d1
·
verified ·
1 Parent(s): 9c1d384

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -8
README.md CHANGED
@@ -14,7 +14,7 @@ metrics:
14
  - mAP50-95
15
  ---
16
  # Description
17
- YOLOs in this repo are trained with datasets that i have annotated myself. YOLOs on open datasets will have their own pages.
18
  #### Want to request a model?
19
  Im open to commissions, hit me up in Discord - **anzhc**
20
  ## Available Models
@@ -50,21 +50,31 @@ Benchmark was performed in 640px.
50
 
51
  ![image/png](https://cdn-uploads.huggingface.co/production/uploads/633b43d29fe04b13f46c8988/T5Q_mPJ8Ag6jfkaTpmNlM.png)
52
 
53
- ### Eyes segmnetation:
54
- PLACEHOLDER
 
55
 
56
  | Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
57
  | --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
58
- | Anzhcs ManFace v02 1024 y8n.pt | Face: real | 0.883(box),0.883(mask) | 0.778(box), 0.704(mask) |1(face) |~340 |1024|
59
- | Anzhcs WomanFace v05 1024 y8n.pt | Face: real | 0.82(box),0.82(mask) | 0.713(box), 0.659(mask) |1(face) |~600 |1024|
 
 
 
 
 
60
 
61
  ### Head+Hair segmentation:
62
- PLACEHOLDER
63
 
64
  | Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
65
  | --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
66
- | Anzhcs ManFace v02 1024 y8n.pt | Face: real | 0.883(box),0.883(mask) | 0.778(box), 0.704(mask) |1(face) |~340 |1024|
67
- | Anzhcs WomanFace v05 1024 y8n.pt | Face: real | 0.82(box),0.82(mask) | 0.713(box), 0.659(mask) |1(face) |~600 |1024|
 
 
 
 
68
 
69
  ### Breasts segmentation:
70
  Model for segmenting breasts. Was trained on anime images only, therefore has very weak realistic performance, but still is possible.
 
14
  - mAP50-95
15
  ---
16
  # Description
17
+ YOLOs in this repo are trained with datasets that i have annotated myself, or with the help of my friends(They will be appropriately mentioned in those cases). YOLOs on open datasets will have their own pages.
18
  #### Want to request a model?
19
  Im open to commissions, hit me up in Discord - **anzhc**
20
  ## Available Models
 
50
 
51
  ![image/png](https://cdn-uploads.huggingface.co/production/uploads/633b43d29fe04b13f46c8988/T5Q_mPJ8Ag6jfkaTpmNlM.png)
52
 
53
+ ### Eyes segmentation:
54
+ Was trained for the purpose of inpainting eyes with Adetailer extension, and specializes on detecting anime eyes, particularly - sclera area, without adding eyelashes and outer eye area to detection.
55
+ Current benchmark is likely inaccurate (but it is all i have), due to data being re-scrambled multi times (dataset expansion for future versions).
56
 
57
  | Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
58
  | --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
59
+ | Anzhc Eyes -seg-hd.pt | Eyes: illustration | 0.925(box),0.868(mask) | 0.721(box), 0.511(mask) |1(eye) |~500(?) |1024|
60
+
61
+
62
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/633b43d29fe04b13f46c8988/o3zjKGjbXsx0NyB5PNJfM.png)
63
+
64
+
65
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/633b43d29fe04b13f46c8988/WIPhP4STirM62b1qBUJWf.png)
66
 
67
  ### Head+Hair segmentation:
68
+ An old model (one of my first). Detects head + hair. Can be useful in likeness inpaint pipelines that need to be automated.
69
 
70
  | Model | Target | mAP 50 | mAP 50-95 |Classes |Dataset size|Training Resolution|
71
  | --------------------------- | --------------------- | ----------------------------- | ------------------------- |---------------|------------|-------------------|
72
+ | Anzhc HeadHair seg y8n.pt | Head: illustration(maybe real?) | 0.775(box),0.777(mask) | 0.576(box), 0.552(mask) |1(head) |~3180 |640|
73
+ | Anzhc HeadHair seg y8m.pt | Head: illustration(maybe real?) | 0.867(box),0.862(mask) | 0.674(box), 0.626(mask) |1(head) |~3180 |640|
74
+
75
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/633b43d29fe04b13f46c8988/Ic2n8gU4Kcod0XwQ9jzw8.png)
76
+
77
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/633b43d29fe04b13f46c8988/oHm-Z5cOPsi7OfhmMEpZB.png)
78
 
79
  ### Breasts segmentation:
80
  Model for segmenting breasts. Was trained on anime images only, therefore has very weak realistic performance, but still is possible.