Update README.md

15d7556 verified 3 months ago

631 Bytes

license: gpl-3.0
base_model:
  - Ultralytics/YOLOv8
pipeline_tag: object-detection

YonkersNet - A Yolo model that can predict how big/small is your waifu chest

this is my late year holiday side project. I built this model just by using 2000 images for each of the model (total 4000 images). Yes, the model is separated into 2 weights because it has different dataset & im using ensemble approach for this time.

The model will predict with this class

0 - small
1 - medium
2 - large
3 - toobig

Dataset come from my art folder and some random danbooru picture with [chest size] + rating:safe