Skip to content

A machine learning model capable of finding and creating a bounding box to identify the lego part.

License

Notifications You must be signed in to change notification settings

Harry-cheong/Legosorter

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 

Repository files navigation

AIWinpetition

A machine learning to find lego bricks

Topic: Miscellaneous

Implementing Our Project

For the project, we unfortunately do not have the time to train a machine learning model capable of identifying all LEGO parts out there. Therefore, We have decided that we will create a scalable AI model that could identify at least one lego part as a proof of concept.

Our Data

image

LEGO PART 32498 Technic Gear 36 Tooth Double Bevel The 3D file can be found inside the synthetic data folder

Dataset

With the blender application, we wrote a script to generate data with a 3D model of a gear.

Dataset: https://drive.google.com/drive/folders/1rO6Gt2YgV3EJcvq5wpVG2PDQylUjP6b-?usp=sharing

Machine Learning model

About

A machine learning model capable of finding and creating a bounding box to identify the lego part.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published