Solved by Amrit_Das in post #2 
              
              
                carrying Inference on a trained image, jsut predicts the class of the image. For example in my case, 
 
 
I read one image at a time. You can easily automate this by writing another script that reads images from a folder one by one, predicts them and then based on the predicted class stores them in …