Not-For-All-Audiences
Create README.md
Browse files
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.
|