Lego classification

Hello everyone,
I 'm looking for a model that I can use for image classification,
I don’t know which architecture I have to select in the page of the models.
My purpose is to reach an accuracy of more than 95 % for a lego sorter machine that uses image classification.
Can you help me with this please?
Kind regard

1 Like

Hi @ffsarrah

If you are looking for a simple image classification model, I recommend following this tutorial: Görüntü sınıflandırması  |  TensorFlow Core

Cheers,

2 Likes

Take a look at
https://www.streamhead.com/3d%20printing/ai/2021/11/01/deep-learning-lego-sorting.html

1 Like

I suggest you look at the links given to you by previous comments first, but if they still don’t work(accuracy is bad), I suggest switching to an object detection model as it both localizes the lego and detects the class of the image

1 Like