Update README.md

README.md

@@ -1,15 +1,18 @@
 # NSFW Segmentation
 
-Multi-head release of three single-task segmentation models targeting NSFW anatomy. Each checkpoint runs independently and produces binary masks for the specified classes.
+Multi-head release of single-task segmentation models targeting NSFW anatomy. Each checkpoint runs independently and produces binary masks for the specified classes.
 
-| File | Task | Classes | Mask mAP@0.5 | Mask mAP@0.5:0.95 |
-| --- | --- | --- | --- | --- |
-| `nsfw-seg-breast.pt` | Breast anatomy | breast, areola, nipple | 0.895 | 0.636 |
-| `nsfw-seg-vagina.pt` | Vagina | vagina | 0.995 | 0.871 |
-| `nsfw-seg-penis.pt` | Penis | penis | 0.995 | 0.975 |
+| File | Backbone | Task | Classes | Mask mAP@0.5 | Mask mAP@0.5:0.95 |
+| --- | --- | --- | --- | --- | --- |
+| `nsfw-seg-breast-s.pt` | YOLO11s | Breast anatomy | breast, areola, nipple | 0.895 | 0.636 |
+| `nsfw-seg-breast-x.pt` | YOLO11x | Breast anatomy | breast, areola, nipple | 0.929 | 0.702 |
+| `nsfw-seg-vagina-s.pt` | YOLO11s | Vagina | vagina | 0.995 | 0.871 |
+| `nsfw-seg-vagina-x.pt` | YOLO11x | Vagina | vagina | 0.995 | 0.918 |
+| `nsfw-seg-penis-s.pt`  | YOLO11s | Penis | penis | 0.995 | 0.975 |
+| `nsfw-seg-penis-x.pt`  | YOLO11x | Penis | penis | 0.995 | 0.987 |
 
 ## Description
-- Backbone: YOLO11s segmentation head (Ultralytics 8.3.204).
+- Backbones: YOLO11s and YOLO11x segmentation heads (Ultralytics 8.3.204).
 - Weights exported as `.pt` checkpoints compatible with `ultralytics>=8.3`.
 - One model per label space; load the checkpoint that matches your target anatomy.
 
@@ -36,7 +39,7 @@ Multi-head release of three single-task segmentation models targeting NSFW anato
 ```python
 from ultralytics import YOLO
 
-model = YOLO("nsfw-seg-breast.pt")  # swap for vagina or penis
+model = YOLO("nsfw-seg-breast-s.pt")  # swap for -x or other anatomy
 results = model.predict("path/to/image.jpg", imgsz=832, conf=0.1)
 ```