Image classification of LEGO parts

Hi ! I'm currently working on a machine to capture images of LEGO on a belt conveyor and classifying theses images with a convolutional neural network (CNN) . I already built the conveyor, captures images and train my model but I my results aren't that great.

I capture the images with a Pi camera module 5MP and run an image detection algo on a Pi 3 B+. The inference is done on my computer and all of this work fine. …

classification computervision image lego