How to do image classification in python