使用Tensorflow Hub进行增量训练以进行图像分类

时间:2019-04-29 17:17:51

标签: python-3.x tensorflow

我正在使用TensorFlow Hub在我自己的班级上进行再培训,他们使用在ImageNet上接受过培训的1000个班级中的一种初始模块架构。 image retraining在我的新数据集上效果很好。但是现在,如果我必须对具有相同标签,只是更多数据的新附加数据集进行增量式重新训练,我该怎么做?我知道我们可以使用存储的最新检查点,并告诉模型从最新检查点重新开始训练。但是在这种情况下,据我所知,没有参数可以做到,因为代码默认情况下默认使用TensorFlow集线器的初始模型作为默认值来开始重新训练。我在TF的官方github上遵循this代码

我已经成功地使用了Inception模块对新图像进行再训练,但是不知道如何使用相同的代码再次对保存的模型进行训练

0 个答案:

没有答案