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 使用raw socket进行TCP SYN扫描实例
这篇文章主要介绍了python 使用raw socket进行TCP SYN扫描实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05Python线程协作threading.Condition实现过程解析
这篇文章主要介绍了Python线程协作threading.Condition实现过程解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03Python使用Selenium WebDriver的入门介绍及安装教程(最新推荐)
这篇文章主要介绍了Python使用Selenium WebDriver的入门介绍及安装教程,本文使用环境为python3.11+win10 64位+firefox浏览器,所以本文使用的浏览器驱动是Firefox的geckodriver ,如果你使用的是其他浏览器,那么选择自己对应的浏览器驱动程序即可,需要的朋友可以参考下2023-04-04
最新评论