When i\'m trying to store a set of images from a dataset in a numpy array for processing them using stratified k folder for splitting dataset it ends up crashin