keras 无法多次调用 model.predict_classes

Question

def predictOne(imgPath):

    model = load_model("withImageMagic.h5")
    image = read_image(imgPath)
    test_sample = preprocess(image)
    predicted_class = model.predict_classes(([test_sample]))
    return predicted_class

我已经训练了一个模型。在这个函数中，我加载我的模型，读取一张新图像，进行一些预处理，最后预测它的标签。

当我运行我的 main.py 文件时，这个函数被调用并且一切顺利。然而，几秒钟后，这个函数将被另一个图像再次调用，我得到这个错误：

'Cannot interpret feed_dict key as Tensor: ' + e.args[0])

TypeError：无法将 feed_dict 键解释为 Tensor：Tensor Tensor("Placeholder:0", shape=(5, 5, 1, 32), dtype=float32) 不是此图的元素。

这个功能只在第一次起作用，这很奇怪。我测试了多张图片并得到了相同的行为。

Windows 10 - 带有 keras 的 tensorflow-gpu

Answer 1

尝试从函数外部的文件中加载模型，并将模型对象作为函数def predictOne(imgPath, model) 的参数。这也会快得多，因为每次需要预测时都不需要从磁盘加载权重。

如果要在函数内部继续加载模型，请导入后端：

from keras import backend as K

然后

K.clear_session() 

在加载模型之前。

Answer 2

class one_model:
    session = None
    graph = None 
    loadModel = None
    __instance = None
    @staticmethod
    def getInstance(modelPath):
        """ Static access method. """
        if one_model.__instance == None:
            one_model.__instance = one_model(modelPath)
        return one_model.__instance
        
    def __init__(self, modelPath):
        self.modelPath = modelPath
        self.session = tf.Session(graph=tf.Graph())
        self.loadOneModel()
            
    def loadOneModel(self):
        try:
            with self.session.graph.as_default():
                K.set_session(self.session)
                self.loadModel = keras.models.load_model(self.modelPath)               
        except Exception as e:
            logging.error(str(e))
            print(str(e))
                        
    def getPredictionOne(self, input_file_path): 
        #Predict the data once the model is loaded
        if self.loadModel is not None and self.session is not None: 
            try:
                image = load_img(input_file_path, target_size=inputShape)
                image = img_to_array(image)
                image = np.expand_dims(image, axis=0)
                image = preprocess(image)
                with self.session.graph.as_default():
                    K.set_session(self.session)
                    preds = self.loadModel.predict(image)
                    return preds
            except Exception as e:
                logging.error(str(e))
        
        return -1


if __name__== "__main__": 
    #First Model 
    data = web.input()
        fileapth = data.imagefilepath  
        modelfilepath = data.modelfilepath
        one_modelObj = one_model.getInstance(modelfilepath)        
        value = one_modelObj.getPredictionOne(fileapth)