Invert colors when plotting a PNG file using matplotlib

Question

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

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')
plt.imshow(cube)
plt.ion()

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

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

Thanks in advance.

Answer 1

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.

Answer 2

Try:

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