opencv DNN模块 +tensorflow

本文介绍了如何使用TensorFlow 0.12.0和Python 3.6创建并保存一个简单的模型到PB文件,然后利用OpenCV 3.4.1的DNN模块在Python和C++中加载并测试PB模型,详细展示了Python和C++的代码实现过程。
摘要由CSDN通过智能技术生成

使用的版本是tensorflow0.12.0 + python3.6,搭建一个简单的模型,用来训练得到pb文件。

opencv3.4.1 + VS2015,用来加载使用pb文件。

训练PD文件。

创建项目,新建save_pb.py。

在该网络中,设置输入结点的名字为:conv1/input,输出结点的名字为:conv1/y,(输入输出结点的名称必须设置)建立网络结构:x*w1*w2=y,w1,w2均为随机数,其中X的尺寸:[1,2],w1:[2,3],w2:[3,1],输出的结点y:[1]。

保存该网络,即保存网络结构,与变量值,即w1和w2两个随机数也保存下来代码如下:

import tensorflow as tf
from tensorflow.python.framework import graph_util
 
 
def save_pb(sess):
    constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['conv1/y'])
    with tf.gfile.FastGFile("D://python//hand//graph.pb", mode='wb') as f:
        f.write(constant_graph.SerializeToString())
 
 
def main():
    with tf.variable_scope(&

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值