There is some code to generate Lego images from Lego 3d files here and use it to train models:

https://github.com/jtheiner/LegoBrickClassification

It is based on this post (which others have mentioned)

https://jacquesmattheij.com/sorting-lego-the-software-side/