如何在图(.pb)中重新训练模型?

时间:2018-07-17 10:48:39

标签: python tensorflow

我将模型保存在图形(.pb文件)中。但是现在该模型不准确,我想开发它。我有要学习的其他数据的图片,但是如果有可能或者它是怎么做的,我没有吗?结果必须是新数据pb图的修改。

2 个答案:

答案 0 :(得分:1)

这是一个好问题。实际上,如果有人可以解释如何做到这一点,那就太好了。但除此之外,我可以对您说,这将导致“灾难性的遗忘”,因此无法解决。您必须再次训练所有数据。

但是无论如何,我也想知道,出于测试原因,特别是对于ssd。

答案 1 :(得分:1)

mozzila / DeepSpeech社区提供了一种从冻结图(.pb)初始化训练的方法。它不会恢复优化器参数,因此需要调整学习率。

您可以在以下位置找到代码

https://github.com/mozilla/DeepSpeech/blob/master/DeepSpeech.py#L1562

希望这会有所帮助!