Full Custom YOLO Detection Model

Model Description

This model is a custom-trained YOLO object detection model for multi-class detection and segmentation tasks on a specialized dataset.

It is trained for fine-grained object detection using bounding box annotations across multiple classes.


Intended Use

  • Object detection on NSFW datasets
  • Bounding box classification for custom classes

Limitations

  • Trained on a custom dataset that was boxed with the ten classes by hand.
  • Performance may degrade on unseen domains or distributions, but in testing on 10k out of database images the error rate was less the 4%

Evaluation Results

Overall Metrics

  • Precision: 0.858
  • Recall: 0.808
  • mAP@50: 0.898
  • mAP@50-95: 0.6156

Per-Class Results

Class Images Instances Precision Recall mAP50 mAP50-95
all 613 1683 0.858 0.809 0.898 0.616
person 233 256 0.829 0.902 0.928 0.762
breast 286 298 0.910 0.884 0.964 0.642
vulva 165 166 0.874 0.777 0.873 0.495
butt 127 131 0.848 0.771 0.895 0.596
male 102 108 0.865 0.474 0.718 0.553
penis 237 269 0.824 0.855 0.903 0.577
anal 145 147 0.936 0.905 0.957 0.704
vaginal 182 184 0.888 0.903 0.952 0.619
blowjob 36 36 0.779 0.778 0.869 0.603
handjob 73 88 0.830 0.841 0.925 0.603

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support