OpenCV cv2 image to PyGame image?

Question

def cvimage_to_pygame(image):
    """Convert cvimage into a pygame image"""
    return pygame.image.frombuffer(image.tostring(), image.shape[:2],
                                   "RGB")

The function takes a numpy array taken from the cv2 camera. When I display the returned pyGame image on a pyGame window, it appears in three broken images. I don't know why this is!

Any help would be greatly appreciated.

Heres what happens::

(Pygame on the left)

Answer 1

In the shape field width and height parameters are swapped. Replace argument:

image.shape[:2] # gives you (height, width) tuple

With

image.shape[1::-1] # gives you (width, height) tuple