記錄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