代码之家  ›  专栏  ›  技术社区  ›  Austin

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

  •  1
  • Austin  · 技术社区  · 7 年前

    根据之前训练的Tensorflow模型,我有以下文件:

    graph.pbtxt , checkpoint model.ckpt-10000.data-00000-of-00001 model.ckpt-10000.index ,及 model.ckpt-10000.meta

    我被告知,这个模型的输入大小是一个密集的大小层 5000 输出是一个密集的sigmoid二元分类,但我不知道中间有多少层(我也不是100%肯定输入大小是正确的)。

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

    0 回复  |  直到 7 年前
    推荐文章