记录tensorflow C++ inference 模型

离线端侧的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.输出结果如下:

 

猜你喜欢

转载自www.cnblogs.com/ChrisInsistPy/p/11493161.html