从pbtxt和ckpts在Keras中复制TF模型

时间:2018-10-30 05:46:16

标签: python-3.x tensorflow keras

我不确定我所掌握的信息是否真的可以实现,但是请告诉我是否是这种情况。

从先前训练过的Tensorflow模型中,我有以下文件:

graph.pbtxtcheckpointmodel.ckpt-10000.data-00000-of-00001model.ckpt-10000.indexmodel.ckpt-10000.meta

有人告诉我,此模型的输入大小为5000大小的密集层,输出为Sigmoid二进制分类,但我不知道中间有多少层。 (我也不是100%肯定输入的大小正确。)

从此信息和相关文件中,是否可以将经过训练的权重的TF模型复制到Keras功能模型中?

(想法是将这个小型密集网络添加到VGG-16的最后一个FC层上,因此这将是我的最终目标。)

0 个答案:

没有答案