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):
这个就是预处理步骤,用这个预处理步骤嵌入到自己先前构造的模型中,也就是预训练模型和自身模型的预处理步骤需要保持一致!!!
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
使用python库xlsxwriter库来输出各种xlsx文件的示例
这篇文章主要介绍了使用python库xlsxwriter库来输出各种xlsx文件的示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2020-09-09Python读取xlsx文件报错:xlrd.biffh.XLRDError: Excel xlsx file;no
这篇文章主要给大家介绍了关于Python库xlrd中的xlrd.open_workbook()函数读取xlsx文件报错:xlrd.biffh.XLRDError: Excel xlsx file;not supported问题解决的相关资料,文中通过图文介绍的非常详细,需要的朋友可以参考下2022-08-08
最新评论