Add additional data augmentation and visualize

[lalalune]

We are kind of blindly just matrix transforming, color swapping and augmenting.

• Implement visualization of current augmentation to verify that it makes sense
• Add noise patterns underneath black as an additional augmentation time
• Add some augmentation that replaces token 0 with one of the colors not used in the puzzle