Prepare your own data set for image classification in Python ML