Sorting 2 Metric Tons of Lego

How do you get the training data to recognize 38000 different shapes of lego?<p>How much training data is it even? I noticed in the video only pretty standard lego blocks are identified.