In color_preprocessing function, should we use the same mean and standard deviation both in training and testing?