tensorflow官方github 预训练模型下载

一、官网链接

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

二、使用步骤

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

图示如下
：

2.下载对应文件

Code对应程序源码

Checkpoint对应预训练模型

三、提示

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

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

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

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

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

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

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