無法將模型文件轉換為.tflite

Question

我已經克隆了這個Image Completion存儲庫，它可以在Windows機器上的python中運行。 托管了用於面部完成和一般完成的模型。 但是，文件沒有擴展名.pb。 如何獲得輸入和輸出張量的名稱？ 我嘗試使用本文重命名帶有.pb擴展名的文件並嘗試在tensorboard中加載，但它給出了屏幕截圖中顯示的錯誤。

我也嘗試使用此答案在python中運行此命令，但輸出中沒有節點。

如何獲取輸入和輸出張量之類的詳細信息，以將該模型轉換為適用於Android的.tflite模型？ 屏幕截圖：
導入模型錯誤
張量板錯誤

Answer 1

存儲庫中提供的文件是檢查點 ，而不是可以轉換為.tflite .pb文件。 從saver.restore()進行的saver.restore()調用可以看出這demo.py 。

要獲取一個pb文件，您將必須構建一個SavedModel 。 生成的文件夾將包含.py文件 ，您可以使用該文件進行轉換（假設它包含TFLite支持的ops）。