search

LynnHaDo / Checkbox-Detection

Checkbox Detection Model for Scanned Documents
https://huggingface.co/spaces/linhdo/checkbox-detector
MIT License
45 stars 2 forks source link

Classification model #10

Open Body123 opened 6 days ago

Body123 commented 6 days ago

Very awesome project! Can you please share the details of how you train a YOLO classification only model? If there is a code for training it, it would be very helpful.

Body123 commented 4 days ago

is there any help?

LynnHaDo commented 3 days ago

Hi @Body123,

Thank you for your interest in my project. It's not too hard to train the classification model--once you have a set of checkboxes divided by types, you can pass the dataset into Ultralytics' classification weights and train it. At least that's how I did it--but there may be better ways to approach this! I don't have the code for it available, unfortunately, but it is no more complex than detection code!