tensorflow官方github预训练模型下载方式
更新时间：2024年02月27日 14:22:47 作者：Dcurry
这篇文章主要介绍了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):
这个就是预处理步骤，用这个预处理步骤嵌入到自己先前构造的模型中，也就是预训练模型和自身模型的预处理步骤需要保持一致！！！
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python3 ffmpeg视频转换工具使用方法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
这篇文章主要为大家介绍了python代码实现小程序的登录案例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步早日升职加薪2022-04-04
这篇文章主要介绍了django框架CSRF防护原理与用法,结合实例形式分析了Django框架CSRF防护的概念、原理、使用方法及相关操作注意事项,需要的朋友可以参考下2019-07-07
这篇文章主要给大家介绍了关于如何将.ipynb文件转换成.py文件的详细步骤,文中通过图文介绍的非常详细,大家基本一看就会,需要的朋友可以参考下2023-07-07
subprocess是Python 2.4中新增的一个模块,它允许你生成新的进程,连接到它们的 input/output/error 管道,并获取它们的返回（状态）码,下面小编就来和大家聊聊它的具体使用吧2023-08-08
pytorch中state_dict()和load_state_dict()函数配合使用可以实现状态的获取与重载，load()和save()函数配合使用可以实现参数的存储与读取，这篇文章主要介绍了Pytorch模型参数的保存和加载,需要的朋友可以参考下2023-03-03
关于numpy.polyfit()与Stats.linregress()方法最小二乘近似拟合斜率对比
这篇文章主要介绍了关于numpy.polyfit()与Stats.linregress()方法最小二乘近似拟合斜率对比,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-04-04
这篇文章主要介绍了python编程开发之类型转换convert用法,结合实例形式分析了Python中常见的数据类型及类型转换convert的具体使用方法,需要的朋友可以参考下2015-11-11
这篇文章详细介绍了在服务器上安装和配置Anaconda3、TensorFlow、PyTorch等深度学习环境的步骤,包括下载、初始化、创建环境、验证安装以及解决一些常见问题2025-01-01
这篇文章主要介绍了Python如何用 matplotlib 绘制柱状图，文章对matplotlib模块中详细学习绘制各种柱状图标相关属性和方法，在遇到需要直观展示离散数据点的差异时，我们可以使用bar()或者barh()绘制美观的图表。具有一定的参考价值,需要的朋友可以参考一下2021-12-12
最新评论