首页 > 编程语言 > 详细

记录tensorflow C++ inference 模型

时间:2019-09-09 19:56:11      阅读:90      评论:0      收藏:0      [点我收藏+]

离线端侧的tensorflow库在工业视觉领域是必不可少的,tensorflow官方从1.10以后就只支持bazel编译tensorflow C++源码,为了能在端侧使用tensorflow来对工业成像建模,则需要应用到tensorflow的windows dll/lib。

1.首先编译出windows tensorflow-gpu 1.10.1的dll 和 lib,教程自查

2.keras训练模型,生成h5文件后转换成tf的.ph文件,也可以直接用tensorflow训练出pb文件

3.在visual studio 2019中调用.pb文件构建图,把opencv转换为tensor后初始化session进行前向计算

4.输出结果如下:

技术分享图片

 

 

记录tensorflow C++ inference 模型

原文:https://www.cnblogs.com/ChrisInsistPy/p/11493161.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!