博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
【TensorFlow系列】【六】多模型部署
阅读量:6222 次
发布时间:2019-06-21

本文共 1311 字,大约阅读时间需要 4 分钟。

hot3.png

TensorFlow的多模型部署,关键在于每个模型拥有一个独立的graph与session,各模型间互不干扰即可。最终直接依据各模型的结果,综合起来做决定。

 

import tensorflow as tfimport numpy as npclass Model:    def __init__(self,meta_path,ckpt_path,out_tensor_name,input_tensor_name):        self.graph = tf.Graph()        #恢复模型        with self.graph.as_default():            self.saver = tf.train.import_meta_graph(meta_path)            self.session = tf.Session(graph=self.graph)        with self.session.as_default():            with self.graph.as_default():                self.saver.restore(self.session,tf.train.latest_checkpoint(ckpt_path))                #获取输入输出tensor                self.out = self.graph.get_tensor_by_name(name=out_tensor_name)                self.input = self.graph.get_tensor_by_name(name=input_tensor_name)    #做预测    def predict(self,image):        result = self.session.run(self.out,feed_dict={self.input:image})        index = np.argmax(result,1)        return index[0]Age_pre = Model(meta_path='',ckpt_path='',out_tensor_name='softmax:0',input_tensor_name='input:0')Gender_pre = Model(meta_path='',ckpt_path='',out_tensor_name='softmax:0',input_tensor_name='input:0')with tf.Session() as session:    image = session.run(fetches='')    age = Age_pre.predict(image)    gender = Gender_pre.predict(image)

 

转载于:https://my.oschina.net/u/3800567/blog/1786556

你可能感兴趣的文章
管理培训笔记
查看>>
SpringMVC上传图片总结(2)--- 使用百度webuploader上传组件进行上传图片
查看>>
numRecordsIn 在哪里实现?
查看>>
http keep-alive 解释
查看>>
Nginx或Apache通过反向代理配置wss服务
查看>>
用双十一的故事串起碎片的网络协议(下)
查看>>
python设计模式之猴子补丁模式
查看>>
Facade外观模式(结构性模式)
查看>>
iconfont字体图标的使用方法--超简单!
查看>>
在.net桌面程序中自定义鼠标光标
查看>>
Java命令学习系列(四)——jstat
查看>>
flash设置里面:您的 Flash 设置会一直保留到您退出 Chrome 为止。
查看>>
CSS再学习 之背景色 背景图片
查看>>
Java常见排序算法之折半插入排序
查看>>
linux如何查看所有的用户和组信息?
查看>>
(原創) C++ Primer 4th與Modern C++ (C/C++)
查看>>
电子书下载:Silverlight 4 Business Intelligence Software
查看>>
Android startActivityForResult()的用法
查看>>
正则域名
查看>>
Delphi中COM自动化对象中使用事件
查看>>