python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > tensorflow官方github预训练模型下载

tensorflow官方github预训练模型下载方式

作者:Dcurry

这篇文章主要介绍了tensorflow官方github预训练模型下载方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

一、官网链接

示例:

https://github.com/tensorflow/models/tree/master/research/slim

二、使用步骤

1.寻找你想要的预训练模型

图示如下:

2.下载对应文件

三、提示

1、一般都是用imagenet上的预训练模型初始化权重,在新的数据集下大多数情况下都有效。

2、输入不一样可以使用预训练模型,只是要注意下,最后全连接层有可能不一样,不一样的话对应着自己的网络模型修改这一层就行

3、如果是图像处理,应采用合适的预处理操作,同时让图像的大小保持一致。

4、去掉一些不需要的层;修改对应名称的层以加载对应的权重

5、预处理步骤在slim/preprocessing文件夹下的 preprocessing_factory.py文件中,打开看得到下载模型对应的预处理方法。

Eg:resnet50对应的是vgg_preprocessing文件,点开这个文件找到def mean_iamge subtraction(imagem, means):

这个就是预处理步骤,用这个预处理步骤嵌入到自己先前构造的模型中,也就是预训练模型和自身模型的预处理步骤需要保持一致!!!

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文