metadata
license: gpl-3.0
Custom trained models for the soon to be released Forbidden Vision ComfyUI custom nodes.
GitHub Repository
📦 Models
All models were trained on a custom dataset of 7k+ images (a mix of SFW/NSFW from Civitai 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 -> generally more robust and reliableForbiddenVision_facedetect_yolov11n.pt→ YOLO Nano fine-tune -> if you need maximum speed and lower vram usage
- Training: Fine-tuned on the YOLOv11 Face base.
- Usage: Lightweight, accurate bounding box detection for faces.
🔵 Face Segmentation (EfficientNet-B0 advprop)
- Base: timm EfficientNet-B0 (advprop, Apache-2.0).
- Training: Besides manually annotated civitai and custom images, the model used an equal split of CelebAMask-HQ
- 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.
⚖️ 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.