如何使用C++爲tensorflow加載檢查點和推理？

Question

我Tensorflow與Python API，並得到了這些檢查點模型文件：

model.ckpt-17763.data-00000-of-00001 
model.ckpt-17763.index 
model.ckpt-17763.meta 

但我希望有一個C/C++共享庫（.so文件）集成到生產時。所以我需要加載這些模型文件並用C++代碼進行推理並編譯爲共享庫。有沒有一些教程或樣本來做到這一點？

Answer 1

你可以編寫C++代碼來加載和使用你的圖形，給出的指令here。

您可以使用文件here在TF資源庫之外創建一個張量流的Cmake項目，然後編譯您的庫。

但是，您仍然需要在.so文件旁邊.ckpt文件，我不知道如何在它們之間插入它們。

有很多關於SO的問題和一些教程（參見this answer中引用的兩個），但是由於張量流變化迅速，它們很快就會過時，並且總是有點難以實現正常工作（完全可行，並且越來越容易）。