The baseline code worked for grayscale images.
We have extended the baseline, so that it works for color images as well.
DATASET - Kodak Color Dataset