记录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.输出结果如下:

 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM