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自动化操作Excel/Word/PDF的实战指南
在现代办公环境中,我们经常需要处理各种文档格式,如Excel表格、Word文档和PDF文件,下面我们就来看看如何使用Python自动化进行相关操作吧2025-09-09
Python SQLite3数据库日期与时间常见函数用法分析
这篇文章主要介绍了Python SQLite3数据库日期与时间常见函数用法,结合实例形式分析了Python连接、查询SQLite3数据以及数据库日期与时间常见操作方法,需要的朋友可以参考下2017-08-08


最新评论