解决keras model 多模型使用的方法

Tensor is not an element of this graph; deploying Keras model

解决办法:
在每个模型实例化时创建新的graph
创建模型之前 添加graph
部分代码

class Inference(object):
    loaded_model = None
    classes_in_keras_format = None
    graph=None
    session=None

    def __init__(self, company_id):
        self.company_id = company_id
        if Inference.classes_in_keras_format is not None:
            return
        self.load_classes()
        Inference.classes_in_keras_format = dict(zip(config.classes, range(len(config.classes))))
        self.set_img_format()
        Inference.session = tf.Session()
        Inference.graph = tf.get_default_graph()
        with Inference.graph.as_default():
            with Inference.session.as_default():
                self.create_model()
                self.load_model()
        
	def predict(self, predict_images):
        with Inference.graph.as_default():
            with Inference.session.as_default():
                predict_images = self.preprocess_input(predict_images)
                out = Inference.loaded_model.predict(np.array(predict_images))

解决思路
在这里插入图片描述

解决多模型方法 graph
https://stackoverflow.com/questions/51127344/tensor-is-not-an-element-of-this-graph-deploying-keras-model?r=SearchResults

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值