Skip to content

Comment on Ask HN: How to train an image recognition AI

Comments

I’ve had good results in the past by fine tuning YOLO for image classification and object detection within images, you can find a rough guide here on how to create a good training dataset and such [0] [1]

YMMV though, ultimately accuracy is going to depend on the quality of the labelled data and your use case

There might be models better suited to your specific needs too, but ultimately you’re always going to need the training dataset

[0]: https://labelstud.io/blog/getting-started-with-image-classif...

[1]: https://docs.ultralytics.com/tasks/classify/

AboutSource Built by g1lg1l

Hackerly is an independent reader for Hacker News, built on the public HN API. Not affiliated with Y Combinator.