Thanks for this very clear and useful package.

When calling visualpriors.representation_transform(image_data, "colorization") I get an error (dimension mismatch) when I pass in an RGB color image.

I know I can get a grayscale image along the lines of 0.3 * red + 0.6 * green + 0.1 * blue, but I wanted to check to see what you used during training so it could matched as closely as possible. Perhaps representation_transform could take care of this preprocessing step for the user to take out the guesswork?