深度学习之格式转换笔记(三):keras(.hdf5)模型转TensorFlow(.pb) 转TensorRT(.uff)格式
环境:
tensorflow1.15,cuda10.0,cudnn7.6.4
将keras训练好保存的.hdf5格式模型转为tensorflow的.pb模型,然后转为tensorrt支持的uff格式。
keras(.hdf5)模型转TensorFlow(.pb)
# h5_to_pb.pyfrom keras.models import load_model
import tensorflow as tf
import os
import os.path as osp
from keras import backend as K# 路径参数
input_path = 'D:/pycharm/facenet/models/'
weight_file = 'resnet50.hdf5'
weight_file_path = osp.join(input_path, weight_file)
output_graph_name = weight_file[:-3] + '.pb'# 转换函数
def h5_to_pb(h5_model, output_dir, model_name, out_prefix="output_", log_tensorboard=True):if osp.exists(output_dir) == False:os.mkdir(output_dir)out_nodes = []for i in range(len(h5_model.outputs)):out_nodes.append(out_prefix + str(i + 1))tf.identity(h5_model.output[i], out_prefix + str(i + 1))sess = K.get_session()from tensorflow.python.framework import graph_util, graph_ioinit_graph = sess.graph.as_graph_def()main_graph = graph_util.convert_variables_to_constants(sess, init_graph, out_nodes)graph_io.write_graph(main_graph, output_dir, name=model_name, as_text=False)if log_tensorboard:from tensorflow.python.tools import import_pb_to_tensorboardimport_pb_to_tensorboard.import_to_tensorboard(osp.join(output_dir, model_name), output_dir)# 输出路径
output_dir = osp.join(os.getcwd(), "trans_model")
# 加载模型
print(weight_file_path)
h5_model = load_model(weight_file_path)
# model.load_weights
h5_to_pb(h5_model, output_dir=output_dir, model_name=output_graph_name)
print('model saved')
TensorFlow(.pb) 转TensorRT(.uff)
只需要安装好TensorRT之后通过终端运行convert-to-uff resnet50.pb即可得到resnet50.uff 。
测试:如果你想实验一下可以根据tensorrt自带的例子来进行实验,过程如下:
在jetson nano上测试tensorRT,tensorrt是当时安装镜像的时候就自带有得,通过import tensorrt和import
uff都能够成功,进行测试得时候我用得是在/usr/src/tensorrt/samples/python文件夹下有很多python的例子,我们以第一个end_to_end_tensorflow_mnist的例子为例,mkdir models python model.py
然后下载数据mnist.npz,并开始训练,完成后在models文件夹下生成lenet5.pb文件
然后在进行格式转换,要将tensorflow的pb文件转化为uff格式的文件,首先找到convert_to_uff文件,看自己用的是哪个版本的python,如果是python3,则在/usr/lib/python3.5/dist-packages/uff/bin文件夹下,如果是python2,则在/usr/lib/python2.7/dist-packages/uff/bin文件夹下,我们在终端中进入end_to_end_tensorflow_mnist,运行以下指令
python3 /usr/lib/python3.6/dist-packages/uff/bin/convert_to_uff.py --input_file models/lenet5.pb
则会在models文件夹中生成lenet5.uff文件,完成转换,在通过sample.py进行测试,出现报错没有安装pycuda,后发现这个库是用于GPU加速的。
如果在jetson nano上安装pycuda不会的同学可以看这篇文章:
https://blog.csdn.net/m0_51004308/article/details/115280360
再次运行python3 sample.py 报错Could not find 9.pgm。
解决:在/usr/src/tensorrt/data/mnist下找到download_pgms.py文件并通过python3运行即可。
后面就通过了,运行sample.py之后出现
Test Case:8
Prediction:8
深度学习之格式转换笔记(三):keras(.hdf5)模型转TensorFlow(.pb) 转TensorRT(.uff)格式相关推荐
- 神经网络与深度学习——TensorFlow2.0实战(笔记)(三)(python输入和输出)
从程序中学习知识点 1. #输出函数可以输出字符串常量 print("hello") #输出数学表达式 print(1+2) #输出变量 print(num) 2. #string ...
- 神经网络与深度学习——TensorFlow2.0实战(笔记)(三)(python运算符和表达式)
从程序中学习知识点 1.算术运算符 #运算符(Operator):完成不同类型的常量.变量之间的运算 #除法运算 / 结果是一个浮点型的精确数的值,与java等其他语言的不同之处 print(7/2, ...
- 神经网络与深度学习——TensorFlow2.0实战(笔记)(三)(python常量、变量和表达式)
从程序中学习知识点 1. #支持6中表达形式 数字 字符串 列表 元组 字典 集合 #数字 整型(正整数 负整数 零) #Python3中的整数可以任意大,而不用担心位数不够而导致溢出的情况 intn ...
- 神经网络与深度学习——TensorFlow2.0实战(笔记)(三)(python语句)
1.if语句 #if语句 x,y = 3,5 if x<y:print("x<y") elif x==y:print("x==y") else:pr ...
- 神经网络与深度学习——TensorFlow2.0实战(笔记)(三)(第一个Python程序)
从程序中了解知识点,有详细的备注 # 判断变量num是否为正数 # 注释语句 是对程序进行说明的语句,在程序运行过程中不被执行 # ctrl+/进行多行注释num=0 # 赋值语句 大小写敏感 可以以 ...
- 能在不同的深度学习框架之间转换模型?微软的MMdnn做到了
Microsoft/MMdnn:深度学习框架随心切换 学习深度学习的各位同学都希望自己的模型能在不同的深度学习框架之间随意转换,比如,斯坦福大学CVGL实验室的Saumitro Dasgupta就做过 ...
- Yann Lecun纽约大学《深度学习》2020课程笔记中文版,干货满满!
关注上方"深度学习技术前沿",选择"星标公众号", 资源干货,第一时间送达! [导读]Yann Lecun在纽约大学开设的2020春季<深度学习>课 ...
- [深度学习] 分布式Pytorch介绍(三)
[深度学习] 分布式模式介绍(一) [深度学习] 分布式Tensorflow介绍(二) [深度学习] 分布式Pytorch介绍(三) [深度学习] 分布式Horovod介绍(四) 一 Pytorc ...
- Keras深度学习实战(2)——使用Keras构建神经网络
Keras深度学习实战(2)--使用Keras构建神经网络 0 前言 1. Keras 简介与安装 2. Keras 构建神经网络初体验 3. 训练香草神经网络 3.1 香草神经网络与 MNIST 数 ...
最新文章
- python 字符串 转 dict
- jae的mongo数据库管理工具(原创)
- Winform中将Bitmap对象通过pictureBox显示图片
- 20145315 《Java程序设计》实验五实验报告
- 阿里再破记录!代表中国企业首次在这项国际比赛中摘得银牌!
- Eclipse编辑器字体大小的设置
- linux下超简单的ntp时间服务器
- python画50个图-python 实现在一张图中绘制一个小的子图方法
- 因为M.2 SSD,不得不装了WINDOWS10
- 分销商城系统源码开发
- ipv4v6双栈技术_什么是IPv6双栈技术
- 720nopenwrt设置打印服务器_打印服务器的ip配置及共享设置
- 问题解决:Ubuntu18.04显示器分辨率不正常
- android killer 连接模拟器,AndroidKiller怎么连接夜游神手机模拟器的方法教程
- html5手机端页面缩放问题的解决
- android在framework层增加自己的service仿照GPS
- php单链表检测有没有环,PHP找出链表中环入口节点步骤详解
- 一个高性能无锁非阻塞链表队列
- 应用程序无法正常启动0x000007b,请单击“确定”关闭应用程序(不要着急,先定位问题)
- Python Flask教程学习02