lego-sorting-machine

My first AI project. Written in Python (OpenCV, Keras, TensorFlow and more).

Generating data set

Camera records LEGO bricks (transported on special conveyor belt), computer process images and saves labeled data.

LEGO bricks images can be also generated using 3D models, which are available online. More informations here: https://github.com/Forlify/lego-images-generator

After training, we can start to recognize LEGO bricks!

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
.idea		.idea
img		img
.gitignore		.gitignore
CreateData.py		CreateData.py
CreateGeneratedData.py		CreateGeneratedData.py
README.md		README.md
RecognizeLego.py		RecognizeLego.py
StackImages.py		StackImages.py
TrainLegoModel.py		TrainLegoModel.py
labels.csv		labels.csv
model_trained.p		model_trained.p
model_trained_10epoch_4bricks.p		model_trained_10epoch_4bricks.p