NSFW-API commited on
Commit
684afc4
·
verified ·
1 Parent(s): 23beaf6

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # NSFW Segmentation
2
+
3
+ Multi-head release of three single-task segmentation models targeting NSFW anatomy. Each checkpoint runs independently and produces binary masks for the specified classes.
4
+
5
+ | File | Task | Classes | Mask mAP@0.5 | Mask mAP@0.5:0.95 |
6
+ | --- | --- | --- | --- | --- |
7
+ | `nsfw-seg-breast.pt` | Breast anatomy | breast, areola, nipple | 0.895 | 0.636 |
8
+ | `nsfw-seg-vagina.pt` | Vagina | vagina | 0.995 | 0.871 |
9
+ | `nsfw-seg-penis.pt` | Penis | penis | 0.995 | 0.975 |
10
+
11
+ ## Description
12
+ - Backbone: YOLO11s segmentation head (Ultralytics 8.3.204).
13
+ - Weights exported as `.pt` checkpoints compatible with `ultralytics>=8.3`.
14
+ - One model per label space; load the checkpoint that matches your target anatomy.
15
+
16
+ ## Intended Use
17
+ - Automatic instance segmentation for NSFW anatomical structures in moderated, research, or medical-support workflows.
18
+ - **Inputs:** RGB images.
19
+ - **Outputs:** Binary masks aligned with the class taxonomy above.
20
+
21
+ ## Data Summary
22
+ - Training data consisted of curated, privately-held NSFW image sets with polygon masks (YOLO segmentation format).
23
+ - Train/validation splits were normalized and merged after preprocessing; metrics reflect held-out validation imagery.
24
+ - Datasets are not included in this release.
25
+
26
+ ## Metrics
27
+ - Evaluated with `yolo segment val` at 832 px resolution, confidence threshold 0.1.
28
+ - Numbers in the table refer to the best checkpoint per task.
29
+
30
+ ## Limitations
31
+ - Models are not a substitute for clinical assessment.
32
+ - Domain shift (lighting, camera quality, demographics) may impact performance.
33
+ - No safety filtering is applied; downstream systems must implement access controls.
34
+
35
+ ## Quickstart
36
+ ```python
37
+ from ultralytics import YOLO
38
+
39
+ model = YOLO("nsfw-seg-breast.pt") # swap for vagina or penis
40
+ results = model.predict("path/to/image.jpg", imgsz=832, conf=0.1)
41
+ ```
42
+
43
+ ## Support
44
+ For integration questions or feedback, open an issue on the hosting repository and mention the checkpoint name in the subject line.