將ckpt文件生成pb文件的詳細過程,並調用pb文件進行模型預測.

前段時間搭建了一個分類網絡模型,而後用本身的數據進行了800epoch’的訓練,最後默認生成了三個ckpt文件.因爲要同時運行幾個網絡,因此打算將這個網絡模型進行固化成pb文件,而後直接調用. 主要包括一下內容:前端 1.查看ckpt模型的輸入輸出張量名稱. 2.將ckpt文件生成pb文件. 3.查看生成的pb文件的輸入輸出節點 4.運行pb文件,進行網絡預測 1.查看ckpt網絡的輸入輸出張量名

>>阅读原文<<