Do not transform image to grayscale in train