AI helps collect different objects from a pile of Lego