Student project: LEGO and Computer Vision

Dear all,

This is an idea for an open-source project:

  • count the number of lego bricks in a lego building
  • and divide per color:

we could start from this dataset:

INPUT (video of the lego building)
OUTPUT: Count of the Lego bricks and division per color

Supported by Computational Design Institute (501c3 nonprofit in California)

Duration: 12months

[Deep Learning Part 1]

  1. Dataset analysis
  2. Recognize LegoBrick
  3. Recognize Different Colour

[Deep Learning Part 2]

  1. Recognize Separate Bricks (The Construction will be make by the same type of Bricks )
  2. Using 6 different bricks, [harder]

  1. Deploy on a device (iPad Pro)
1 Like

Making some progress on the research side, I have found an interesting research paper that would be nice to translate to pytorch and fastai as a project. It uses fully connected deep network and MLP. Neural Radiance Fields of View

here the code

and here the video


This is cool. I’m interested. What is your final aim though? to build 3D models of the construction like the nerf paper or just counting bricks during construction?

1 Like

The aim is to achieve both possibly.
I would be personally happy if we arrive to an approximation of the count. But it will be a great to achieve a full 3d model. I am a newbie, but pretty passionate about the subject and with 1 year and half head to develop this project. It will be crucial also to try the integration with Pytorch3D @nchukaobah.

I’m unfamiliar with Pytorch3D. I’ll check it up later this week. I’ll also check out the nerf paper but that might have to wait for next week.

Have you done any work on identifying the blocks? I can work on that this week, maybe put up a github page to chart my progress.I’m thinking, with the dataset, a typical object detection model can work. Or even simple color detection algorithms if color is a strong feature for a block. What do you think?

Thanks @nchukaobah we can use this repository as a starting point, so we can tailor with the initiative of the Computational Design Insitute as a non-profit. Looking forward to work on this with all of you.

Hey @Albertotono, I’m so busy right now so I won’t be able to keep up with the project. Have you been able to make progress?

@nchukaobah, @Albertotono, you might check out the LEGO sorter project that @pryb posted. It’s a very cool project and you might find some interesting inspirations there.

1 Like