plt.imshow TypeError Dimensions non valides pour les données d'image

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