Generated image+label sets should not necessarily be float pixels

[jpsember]

We can save memory by a factor of 4 for example if we are saving rgb images as bytes instead of floats.