我在哪里可以下载tensorflow.contrib.slim.nets的预训练权重?

时间:2018-04-14 21:11:14

标签: tensorflow transfer-learning

我在网上看到了各种各样的地方,甚至还有家庭作业的代码,这些代码似乎知道可以下载各种tf-slim模型的预训练检查点。

示例:http://download.tensorflow.org/models/inception_resnet_v2_2016_08_30.tar.gz

但是如果您访问download.tensorflow.org,则不会返回任何页面,以便您可以选择或只知道下载检查点文件的完全限定URL。

因此,如果您不知道文件的名称(或您想要使用的架构),那么如何找到这些预训练的重量检查点文件?

1 个答案:

答案 0 :(得分:0)

您必须从tensorflow/models存储库开始浏览。当您打开它时,向下滚动,您会看到README.md。它链接到四个模型类别:官方,研究,smaples,教程。

让我们来看看official。它有三个网,mnist,resnet和wide_deep。点击resnet,向下滚动到预训练模型部分,您会在此处找到预训练模型的链接。注意:并非所有型号都有预先训练过的重量。