Classification of complex local environments in systems of particle shapes through shape-symmetry encoded data augmentation