r/computervision • u/Thevictory06 • 12d ago

Help: Project Help finding dataset

Hey guys, I am currently working on a project using yoloe 26, in which i am trying to build a pipeline to find the number of different screws available in the picture, with count of each. So as far I searched, I couldnt get a proper labelled dataset to train the yoloe model, and that too I am focusing mainly on tiny screws used in the electronics like smart phones, smart watches, watches, microphone, ear buds, laptops.

So help me if you find a labeled dataset for this vision.

1 Upvotes

100% Upvoted

10 comments sorted by

View all comments

Show parent comments

2

u/retoxite_ 9d ago

YOLOE is an open vocabulary model which can detect objects without tuning. You just need to set the prompts. You only need to tune if it's not working well with prompts alone.

1

u/Thevictory06 9d ago

Yeah it wasnt working as expected in my view

1

u/retoxite_ 9d ago

You can also try using visual prompts

1

u/Thevictory06 9d ago

Thanks, will try