離線端側的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.輸出結果如下: