2016-09-25 51 views
3

我試圖通過使用預先訓練的網絡來提取新數據集的特徵,例如Google在張量流(發佈之初)中發佈的一個classify_image_graph_def.pb -2015-12-05.tgz)。我在這方面取得了成功,因爲在transfer_learning上有教程,該教程使用classify_image_graph_def.pb(inception_v3.pb)提取新數據集的裂縫。預訓練的檢查點(.chkpt)與GraphDef(.pb)

但是,在新版本的預訓練模型中,tensorflow提供了檢查點文件(例如resnet_v1_152.ckpt)而不是Graph_def(例如resnet_v1_152.pb)。我想知道如何使用這些檢查點文件來提取功能,如transfer_learning。任何人都可以給我一些指導嗎?

回答

0

只需按照官方模型保存/恢復文檔here

相關問題