plt.imshow typeerror invalid dimensions for image data code example

Example: plt.imshow typeerror invalid dimensions for image data

if image.shape[2] == 1:
    image = image[:, :, 0]
return image