Python - Reduciendo la información de una imagen con PCA

import cv2

from matplotlib import pyplot as plt

import numpy as np

from sklearn.decomposition import PCA

import random

img = cv2.imread('coca_cola.jpg')

graycola = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

rows, cols, bands = img.shape

num_max_comp= min(cols,rows)

for n in range (1,5):

     m=n**2

     pca = PCA(n_components=m)

     cola_pca =pca.fit_transform(graycola)

     rcola_pca = pca.inverse_transform(cola_pca)

     plt.figure()

     plt.title('Nº componentes %d' %m);

     plt.imshow(rcola_pca,cmap='gray');