在我的一個項目中,我使用了一個公開的預先訓練的啓動-v3模型:http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz。在tensorflow中使用inception-v3檢查點文件
我只想使用最後一個特徵矢量(pool_3/_reshape:0的輸出)。通過查看腳本示例classify_image.py,我可以成功將圖像傳遞到Deep DNN,提取瓶頸張量(bottleneck_tensor = sess.graph.get_tensor_by_name('pool_3/_reshape:0')
)並將其用於更多目的。
我最近看到有一個更新的訓練初始模型。培訓檢查點位於此處:http://download.tensorflow.org/models/image/imagenet/inception-v3-2016-03-01.tar.gz。
我想用這個新的pretrained而不是舊的。但是文件格式不同。 「舊模型」使用ProtocolBuffer形式的圖形def(classify_image_graph_def.pb),它很容易重用。 「新的」只提供了一個檢查點格式,而我正努力將其插入到我的代碼中。
是否有一種簡單的方法將檢查點文件轉換爲可用於創建圖形的ProtocolBuffer文件?
在此先感謝!
你已經採取了看看這個教程? https://www.tensorflow.org/programmers_guide/variables#checkpoint_files它解釋瞭如何從檢查點文件恢復。 – Neal