Image classification using cnn
I haven't seen a library or module that will do this for you, but it should be simple enough to handle. If you saved the image names in your original dataset while loading, you can add the predictions to that same dataset and then loop through row by row, renaming the image based on the classification category.