会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
卡冈图雅
博客园
首页
新随笔
联系
订阅
管理
2020年5月21日
TensorRT推理加速-基于Tensorflow(keras)的uff格式模型(文件准备)
摘要: 一、引子//Windows tf(keras)训练好了模型,想要用Nvidia-TensorRT来重构训练好的模型为TRT推理引擎加快推理的速度。 二、准备文件 1、训练好模型以后(keras)可以通过以下方式保存keras模型为h5文件 tf.keras.models.save_model(mod
阅读全文
posted @ 2020-05-21 18:51 卡冈图雅blackhole
阅读(4061)
评论(0)
推荐(0)
编辑
公告