Tensorflow对象检测:在本地运行时使用传输学习

时间:2017-06-19 02:53:04

标签: tensorflow deep-learning

从Tensorflow文档中,当您从cloud运行时,我们可以使用转移学习进行对象检测。另外,我们可以使用转移学习在本地运行吗?我看到我们有一个关于在local上运行的文档,但是在本地机器上运行时,我找不到任何关于转移学习的文档。我在pipeline期待它,但我找不到它。那么我们如何在本地运行时配置转移学习呢?

1 个答案:

答案 0 :(得分:5)

你是对的。您可以在本地运行时使用转移学习。

这些步骤类似于running pets on google cloud上的说明,但您的训练配置应该引用您的本地文件系统而不是GCS上的路径。特别是,您需要配置train_config.fine_tune_checkpoint字段以指向解压缩的检查点。

有关详细信息,请参阅sample resnet101 config。您只需要更改PATH_TO_BE_CONFIGURED字符串以指向正确的值。

相关问题