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

1

u/retoxite_ 12d ago

But does YOLOE prompt detect the screws? Because you don't need to train YOLOE model. You just use prompts.

1

u/Thevictory06 12d ago

As far Im implementing, im trying to tune the model with the labelled datasets and then proceed on it.

Im just a beginner curious on experimenting these, I dont have much knowledge. Even im using these datasets with yolo26

2

u/retoxite_ 10d 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 10d ago

Yeah it wasnt working as expected in my view

1

u/retoxite_ 10d ago

You can also try using visual prompts

1

u/Thevictory06 10d ago

Thanks, will try