I would just like to get the class names of the predictions. I can get the class names on the images that I trained the model. But if I predict an image (say which is not trained but already belongs to the pre-trained model (VGG16) I cannot get its class name).
My scenario: I used VGG16 pre-trained and added new datasets (like logos, stars, hills etc which are not present in VGG16 (say) )
Now I trained this model in those datasets, when i predicted an image, it gave the correct class-name (logos, stars, hills etc..)
But when I predict an image that was already in VGG16 (say Dog, Cat), it is not giving me the right class name i.e Dog, Cat etc.
I need some help in this scenario.
Is there any way to get the get the class name?