image classification using python