Skip to content

In this project I create a CNN to clasify lego images from a kaggle dataset. These lego peices were rendered in 3D using Blender. I use data augmentation with a custom built CNN and Keras Callbacks to acheive a 96% accuracy. I used Kaggle's cloud GPUs to run my model and load my data.

Notifications You must be signed in to change notification settings

Terrance-Whitehurst/Keras-Lego-Parts-Classifcation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 

Repository files navigation

Project Overview

In this project I create a CNN to clasify lego images from a kaggle dataset. These lego peices were rendered in 3D using Blender. I use data augmentation with a custom built CNN and Keras Callbacks to acheive a 96% accuracy. I used Kaggle's cloud GPUs to run my model and load my data.

Dataset:

https://www.kaggle.com/joosthazelzet/lego-brick-images

Contains 6400 images of 16 different Lego bricks. The bricks are classified by folders. The images are computer rendered using Blender. For more info please contact me.

About

In this project I create a CNN to clasify lego images from a kaggle dataset. These lego peices were rendered in 3D using Blender. I use data augmentation with a custom built CNN and Keras Callbacks to acheive a 96% accuracy. I used Kaggle's cloud GPUs to run my model and load my data.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published