Creating input for image classification