๐ฏ BreastBox Detection (YOLOv8s+)
The Precision Tool for Breast Detection in Stable Diffusion
BreastBox is a custom-trained object detection model (YOLOv8s+) designed specifically to solve common detection failures in ADetailer. While default models often struggle with side views, overlapping clothing, or dynamic poses, BreastBox provides a tighter, more anatomically correct bounding box.
๐ฅ Why Use BreastBox?
โ High Precision: Manually curated dataset, moving from 200 to over 650 high-quality labels.
โ Challenging Angles: Expertly handles "lying down", "side-boob", "from behind", and partially obscured views.
โ Anatomically Correct: Designed for tighter masks that follow the natural shape of the body.
โ Style Agnostic: Optimized for 3DCG (DAZ/Poser), but generalizes well to realistic and 2D/Anime styles.
๐ Version History
v1.0 (Experimental): Initial release (~200 samples). Best for standard poses.
v2.0: Major data upgrade (~400 samples). Better versatility for breast sizes and camera angles.
v3.0+ (Latest): Focus on obscured breasts (hands, objects) and smaller chest sizes (~700+ samples).
๐ Quick Start (ADetailer)
Download the
.ptfile.
Move it to:
stable-diffusion-webui\models\ADetailerSelect
BreastBoxfrom the ADetailer dropdown in WebUI.
Recommended Settings:
Detection Confidence:
0.3to
0.5
Mask Merge Mode:
Merge
๐ ๏ธ The Power Behind the Training: YOLO Training GUI
I trained and refined this model using YOLO Training GUI, a standalone tool I developed to make YOLO training visual and simple for everyone.
If you are training your own models and want to see your results like I do, check it out: ๐ YOLO Training GUI on GitHub
Why you'll love it:
Live Monitor: Watch detections happen in real-time as the model trains.
Side-by-Side Comparison: Generates the dual-view comparison images.
No Code Required: A full GUI to manage your datasets, hyperparameters, and exports.
๐ฌ Feedback & Community
This is a passion project! If you encounter edge cases or have suggestions:
Share your Feedback in the comments.
Post your Example Images (it helps me refine the dataset!).
If you like the model or the training tool, a Star on GitHub or a review here is much appreciated! โญ
Description
Initial version.
FAQ
Comments (4)
This is strictly a bounding box, not a segmentation mask, right? If so, would you consider adding a segmentation variant?
Yes, correct, this model strictly provides bounding boxes.
Regarding a segmentation variant :
I haven't planned one for now because annotating high-quality masks for breasts is extremely time-consuming compared to boxes.
Also, segmentation can be tricky with fine details like nipple piercings (especially dangling/loose jewelry), where masks often fail or become inaccurate. Bounding boxes are usually robust enough for ADetailer/inpainting workflows.
use a sam(2/2.1/3) model it will segment and mask whatever the bbox detects
@eloraKย Yeah, fair.










