k fold cross validation python code for image classification