Update README.md

README.md

@@ -1,6 +1,37 @@
 ---
 license: gpl-3.0
 ---
+
+<div align="center">
+
 <img src="images/header.webp" width="800px" />
 
-Custom trained models for the Forbidden Vision comfyui custom nodes.
+Custom trained models for the **Forbidden Vision** ComfyUI custom nodes.
+
+</div>
+
+---
+
+## 📦 Models
+
+### 🟢 Face Detection (YOLO fine-tunes)
+- **Base:** YOLO face detection models (GPL-3.0).  
+- **Variants:**  
+  - `ForbiddenVision_facedetect_yolov11s.pt` → YOLO Small fine-tune.  
+  - `ForbiddenVision_facedetect_yolov11n.pt` → YOLO Nano fine-tune.  
+- **Training:** Fine-tuned on mixed anime + real faces.  
+- **Usage:** Lightweight, accurate bounding box detection for faces.  
+
+---
+
+### 🔵 Face Segmentation (EfficientNet-B0 advprop)
+- **Base:** timm EfficientNet-B0 (advprop, Apache-2.0).  
+- **Training:** Fine-tuned for binary face/background segmentation.  
+- **Usage:** Provides clean masks for downstream processing.  
+
+---
+
+### 🟣 Landmark Rotation (MobileNetV3-Small)
+- **Base:** MobileNetV3-Small (Apache-2.0).  
+- **Training:** Trained for landmark-based rotation correction.  
+- **Usage:** Aligns faces to level eyes for consistent preprocessing.