如何在没有源代码的情况下修改预训练的张量流模型?

时间:2018-07-03 20:34:49

标签: tensorflow

我想修改一个预先训练的模型(具体来说,是张量流检测动物园中的ssdlit-mobilenetv2)。 我想做的是提取一层的输出,对输出进行一些修改,然后将其反馈并继续运行模型。 我只有.pb文件和ckpt文件,没有源代码怎么办?

那我应该使用什么代码?

1 个答案:

答案 0 :(得分:0)

我认为,您没有做到这一点的简便方法。只需要根据现有数据编写自己的工具进行具体修改,然后再将其另存为检查点并从该点继续训练即可。

如果文件包含值的信息,那么您可以手动更改它,但我认为它可能无效。

此外,正如您所知道的名称(mobilentv2),您可以找到网络体系结构和源代码