You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

如何获取tensorflow.contrib.slim.nets系列模型的预训练权重?

解决TF-Slim预训练检查点下载URL无法获取的问题

我经常碰到开发者遇到这个困扰——download.tensorflow.org确实没有提供可浏览的文件列表页面,要是不清楚模型对应的文件名和日期后缀,确实很难定位到正确的预训练检查点下载链接。这里分享几个实用的解决办法:

  • 利用模型命名规律拼接URL
    TF-Slim的预训练模型命名有固定规律:[模型架构名]_YYYY_MM_DD.tar.gz,比如你提到的inception_resnet_v2_2016_08_30.tar.gz就是典型格式。常见的TF-Slim预训练模型及对应URL格式如下:

    • Inception系列:http://download.tensorflow.org/models/inception_v1_2016_08_28.tar.gzhttp://download.tensorflow.org/models/inception_v3_2016_08_28.tar.gz
    • ResNet系列:http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gzhttp://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
    • MobileNet系列:http://download.tensorflow.org/models/mobilenet_v1_1.0_224_2017_06_14.tar.gz
  • 查看TF-Slim仓库的历史归档
    TensorFlow官方的models仓库中,slim模块的旧版README里会完整列出所有预训练模型的下载链接。你可以在GitHub上搜索该仓库的历史提交记录,找到包含完整模型列表的README归档版本,从中获取准确的URL。

  • 用命令行工具验证URL有效性
    如果你不确定某个模型的日期后缀,可以用wgetcurl的蜘蛛模式快速验证URL是否存在,不会实际下载文件:

    wget --spider http://download.tensorflow.org/models/[你猜测的模型文件名].tar.gz
    

    或者:

    curl --head http://download.tensorflow.org/models/[你猜测的模型文件名].tar.gz
    

    如果返回200 OK,就说明这个URL是有效的。

  • 通过TensorFlow Hub间接获取
    很多TF-Slim模型也托管在TensorFlow Hub上,你可以在Hub中搜索对应模型,部分模型的详情页会提供原始预训练检查点的下载链接,或者你可以从Hub的模型加载逻辑中反向推导正确的下载路径。

内容的提问来源于stack exchange,提问作者Joey Carson

火山引擎 最新活动