Invert colors when plotting a PNG file using matplotlib


I'm trying to display a PNG file using matplotlib and of course, python. For this test, I've generated the following image: Original File

Now, I load and transform the image into a multidimensional numpy matrix:

import numpy as np
import cv2
from matplotlib import pyplot as plt

cube = cv2.imread('Graphics/Display.png')

When I try to plot that image in matplotlib, the colors are inverted:Plot

If the matrix does not have any modifications, why the colors in the plot are wrong?

Thanks in advance.

1/17/2013 8:36:28 PM

Accepted Answer

It appears that you may somehow have RGB switched with BGR. Notice that your greens are retained but all the blues turned to red. If cube has shape (M,N,3), try swapping cube[:,:,0] with cube[:,:,2]. You can do that with numpy like so:

rgb = numpy.fliplr(cube.reshape(-1,3)).reshape(cube.shape)

From the OpenCV documentation:

Note: In the case of color images, the decoded images will have the channels stored in B G R order.

1/17/2013 9:21:08 PM


plt.imshow(cv2.cvtColor(cube, cv2.COLOR_BGR2RGB))

Licensed under: CC-BY-SA with attribution
Not affiliated with: Stack Overflow