我想修改一个预先训练的模型(具体来说,是张量流检测动物园中的ssdlit-mobilenetv2)。 我想做的是提取一层的输出,对输出进行一些修改,然后将其反馈并继续运行模型。 我只有.pb文件和ckpt文件,没有源代码怎么办?
那我应该使用什么代码?
答案 0 :(得分:0)
我认为,您没有做到这一点的简便方法。只需要根据现有数据编写自己的工具进行具体修改,然后再将其另存为检查点并从该点继续训练即可。
如果文件包含值的信息,那么您可以手动更改它,但我认为它可能无效。
此外,正如您所知道的名称(mobilentv2),您可以找到网络体系结构和源代码