[SOLVED] Keras 2: Flow_from_directory messing up colors

[KevinB (Kevin Bird)]

Just recently @goyder helped me fix that same issue. Really frustrating issue, but at least it is an easy solution!

Breaking Down get_data() Function

Hey Kevin! I’ve just spent the last twenty minutes working through the same problem you’ve just had. The key thing that’s missing for displaying the image is casting the NumPy array to a uint8. Instead of plt.imshow(data[0][0]) try plt.imshow(data[0][0].astype(np.uint8)). Seems that generated image is in a float64 format, which appears to be fine for Keras, but not for Matplotlib. You’ll also note that this conversion is done under Jeremy’s ‘plots’ function under utils.py. Obligatory SO …