block2.py

# Modules used by the function
#import numpy as np
#from src.commons import utils 
#from constants import  

#   Stack images
pixel_matrix = None
fitted_model = None
for idx, row in data.iterrows():
    # Load the image using OpenCV and converting from BGR to RGB
    img = utils.load_image('{}/{}.jpg'.format(IMAGE_RESHAPED_FOLDER, row.title)) 
    reshaped = img.reshape((-1, 3))

    if pixel_matrix is None:
        pixel_matrix = reshaped
    else:
        pixel_matrix = np.concatenate((pixel_matrix, reshaped))