Update README.md

README.md

@@ -6,30 +6,37 @@ license: gpl-3.0
 
 <img src="images/header.webp" width="800px" />
 
-Custom trained models for the soon to be releasedd **Forbidden Vision** ComfyUI custom nodes.
-https://github.com/luxdelux7/ComfyUI-Forbidden-Vision
-  <a href="https://ko-fi.com/luxdelux" target="_blank">
-    <img src="https://ko-fi.com/img/githubbutton_sm.svg" alt="Support me on Ko-fi">
-  </a>
+Custom trained models for the soon to be released **Forbidden Vision** ComfyUI custom nodes.  
+<a href="https://github.com/luxdelux7/ComfyUI-Forbidden-Vision">GitHub Repository</a>  
+
+<br/>
+
+<a href="https://ko-fi.com/luxdelux" target="_blank">
+  <img src="https://ko-fi.com/img/githubbutton_sm.svg" alt="Support me on Ko-fi">
+</a>
+
 </div>
 
 ---
 
 ## 📦 Models
 
+All models were trained on a custom dataset of **7k+ images** (a mix of SFW/NSFW portraits from public sources and my own curated images).  
+The goal was to cover style diversity across anime and realistic domains.
+
 ### 🟢 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 the <a href="https://github.com/YapaLab/yolo-face">yolo 11 face base</a> using a custom dataset of 7k+ images which were a mix of civitai sfw and nsfw content + my own curated images
+- **Training:** Fine-tuned on the <a href="https://github.com/YapaLab/yolo-face">YOLOv11 Face base</a>.  
 - **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.  
+- **Training:** Besides manually annotated civitai and custom images, the model used an equal split of <a href="https://github.com/switchablenorms/CelebAMask-HQ">CelebAMask-HQ</a>
 - **Usage:** Provides clean masks for downstream processing.  
 
 ---
@@ -38,3 +45,9 @@ https://github.com/luxdelux7/ComfyUI-Forbidden-Vision
 - **Base:** MobileNetV3-Small (Apache-2.0).  
 - **Training:** Trained for landmark-based rotation correction.  
 - **Usage:** Aligns faces to level eyes for consistent preprocessing.  
+
+---
+
+⚖️ **Licensing Note**  
+This repository is distributed under **GPL-3.0** due to the YOLO fine-tunes.  
+The segmentation and rotation models are originally Apache-2.0 based, but included here under GPL for consistency.