使用tensorflow源代码示例中提供的retrain.py脚本,我重新培训了inception-v3模型的顶层,以便能够对花集进行分类(提供了5朵花的图像集)。
我现在要做的是,采用新的迁移学习模式(TL-模式),然后用两组新的花来扩展它,而不必完全重新训练整个模型。(5种花卉-> 7种花卉)
我的方法:
inception-v3)替换为TL模型,并尝试在启动时构建该模型。这导致了张量和操作在顶层的各种重复,以及softmax子图的丢失。我觉得我从根本上遗漏了一些东西,从错误的方向处理这个问题。任何洞察力都会受到极大的赞赏。
发布于 2017-12-05 16:43:05
(根据@dga的请求,从评论转换为答复)
不幸的是,如果不对模型结构做一些“脑外科手术”,这并不容易。我没有一个很好的答案,因为我自己从来没有这样做过,但从理论上讲,您应该能够将旧的五个类别完全连接的层恢复到一个新的扩展的七类版本中,然后重新运行培训。
https://stackoverflow.com/questions/41231526
复制相似问题