Solved by manelpk in post #4 
              
              
                Found the solution! Here it is, in case someone finds it useful in the future. 
I was splitting the data into train and test sets incorreclty (due to a wrong intepretation on how numpy arrays work), so I decided to change the approach by using list comprehensions: 
training_images = [images_path[i] …