inception version 4 model, saving and loading the

2019-08-23 19:58发布

问题:

I am kind of new to deep learning and trying to do something related to image classification. I have managed to retrain inception_v4 model by modifying the retrain.py code that was provided for the inception_v3 model. Getting a prediction in the testing stage of the same file works, but when I am trying to get a prediction from it after loading the model from a saved graph def, there are some errors in input image. Can some help show me how to save and load the inception v4 model after retraining... all the from saving the graph def upto getting prediction from it?