IIRC the idea is not to identify every single shape, but to sort in a few categories that are easier to resell, like "yellow", "hi-tech", "heads" (I don't remember tthe exact categories.)
You can try to get more info in the original threads posted by the author:
How much training data is it even? I noticed in the video only pretty standard lego blocks are identified.