使用VGG16.npy文件载入权重
VGG16.npy文件的用法
- VGG16网络权重直接载入
- VGG16.npy加载方式
- 预先把网络结构设置好
- 载入VGG16.npy权重文件
- 了解VGG16.npy内部存储模型权重层的名称
- 按层数逐层给模型权重层载入权重系数
- 存储VGG16模型和权重
VGG权重(npy格式)百度网盘链接: link
提取码:8487
VGG16网络权重直接载入
因为tensorflow包在安装时就已经自带了VGG16、AlexNet、VGG19等模型的权重载入路径,所以通常情况下我们要使用一些预训练的网络时可以直接利用tensorflow下载。
import tensorflow as tf
model=tf.keras.applications.vgg16.VGG16(weights=True,include_top=True) #带全连接层权重
但这种方式基本很难下载,因为要进外网,国内下载速度太慢甚至根本下载不了,即使直接去相应的官网也是一样。
VGG16.npy加载方式
想要连外网下载像VGG这类模型权重也不太容易,VGG16模型权重系数527M,去除顶层(全连接层)权重系数50M左右,所以我推荐的是找一些网盘文件等,这样也比较方便,只是网盘的下载速度可能有点慢,但费点时间总是可以完成的。这里我给出了VGG16模型的权重下载链接,不过是npy文件格式,载入权重过程如下。
预先把网络结构设置好
import tensorflow as tf
import numpy as py
model=tf.keras.applications.vgg16.VGG16(weights=None,include_top=True) #把VGG16网络的结构设置好,此时网络没有权重,类似一个空壳
载入VGG16.npy权重文件
data_dict=np.load('D:/vgg16.npy',encoding='latin1',allow_pickle=True).item()#这是一个储存VGG16权重的字典文件
了解VGG16.npy内部存储模型权重层的名称
for key in data_dict:print (key)
输出结果如下:
conv5_1
fc6
conv5_3
conv5_2
fc8
fc7
conv4_1
conv4_2
conv4_3
conv3_3
conv3_2
conv3_1
conv1_1
conv1_2
conv2_2
conv2_1
这样我们就获得了VGG16.npy内部存储的模型各权重层名称(这里是防止有些模型权重存储时层的名称被修改),然后通过这些名称给对应的模型权重层载入权重系数。
按层数逐层给模型权重层载入权重系数
model.layers[1].set_weights(data_dict['conv1_1'])
model.layers[2].set_weights(data_dict['conv1_2'])
model.layers[4].set_weights(data_dict['conv2_1'])
model.layers[5].set_weights(data_dict['conv2_2'])
model.layers[7].set_weights(data_dict['conv3_1'])
model.layers[8].set_weights(data_dict['conv3_2'])
model.layers[9].set_weights(data_dict['conv3_3'])
model.layers[11].set_weights(data_dict['conv4_1'])
model.layers[12].set_weights(data_dict['conv4_2'])
model.layers[13].set_weights(data_dict['conv4_3'])
model.layers[15].set_weights(data_dict['conv5_1'])
model.layers[16].set_weights(data_dict['conv5_2'])
model.layers[17].set_weights(data_dict['conv5_3'])
model.layers[20].set_weights(data_dict['fc6'])
model.layers[21].set_weights(data_dict['fc7'])
model.layers[22].set_weights(data_dict['fc8'])
这样载入完成后的model就是我们需要的VGG16模型了。
存储VGG16模型和权重
为了方便以后调用VGG16模型,我们可以对其进行存储,存储方式如下。
model.save("VGG16.h5") #这样就把VGG16模型存储在当前路径下了
model.save_weights("VGG16_weights.h5")#把VGG16文件模型权重系数存储在当前路径下,大小约为527M
这样就完成了VGG16.npy文件的载入和重新存储。对于其他模型权重的npy格式文件,载入方式也可以按这个来。因为当前许多经典的卷积神经网络模型直接下载不易,网上能找到的基本也是npy格式的模型权重文件,所以这里给出了一个加载这类npy权重文件的方法,希望能减少大家的弯路。
使用VGG16.npy文件载入权重相关推荐
- python npy文件_python实现npy格式文件转换为txt文件操作
如下代码会将npy的格式数据读出,并且输出来到控制台: import numpy as np ##设置全部数据,不输出省略号 import sys np.set_printoptions(thresh ...
- python如何打开npy文件_操作python实现npy格式文件转换为txt文件
如下代码会将npy的格式数据读出,并且输出来到控制台:import numpy as np ##设置全部数据,不输出省略号 import sys np.set_printoptions(thresho ...
- python如何打开npy文件_python实现npy格式文件转换为txt文件操作
如下代码会将npy的格式数据读出,并且输出来到控制台: import numpy as np ##设置全部数据,不输出省略号 import sys np.set_printoptions(thresh ...
- pytorch中保存、载入权重
有两套模板,配套使用: 模板一:直接加载参数模型 # 保存权重 torch.save(model, save_dirt) # 载入模型 model = torch.load(save_dirt) ...
- npy文件的打开,读取
在复现Sketchmate的时候,碰到了一个叫npy文件的东西: alpha = 0.001 cf_class = torch.from_numpy(np.load("center_feat ...
- .npy文件存储的字典转为json文件
首先,如果你的数据需要经常打开浏览的话,请不要把文件存成这个格式!!! npy文件的格式是存储成二进制格式的,应该是比较适合存储数值型的文件.并且你不能像csv那样即时地打开修改(读取只能用numpy ...
- 什么是npy文件,为什么要用npy格式保存文件?
文章目录 一.官方文档(中文) 二.官方文档(英文) 三.为什么要用npy文件保存文件 四.读取和保存 (1)加载npy文件,并将npy文件写入一个txt文件 (2)保存npy文件 (3)结构解析 一 ...
- .npy文件_python如何利用numpy存取文件
NumPy提供了多种存取数组内容的文件操作函数.保存数组数据的文件可以是二进制格式或者文本格式.二进制格式的文件又分为NumPy专用的格式化二进制类型和无格式类型. numpy格式的文件可以保存为后缀 ...
- 利用Python模拟鼠标操作,一键将GIF文件载入CSDN的Markdown编辑器
简 介: 讨论了在csdn的Markdown编译器中自动载入GIF文件的方法.使用了Python中的pyautogui软件包,可以通过Python的小程序实现一键将GIF文件载入CSDN的Markdo ...
- 上传镜像文件到服务器,通过把docker镜像保存为文件载入到别的服务器
在我们的工作环境中,一般镜像通过上传到镜像仓库中(本地或者公共仓库),用到时下载到服务器上即可.如果服务器不能上网,又没有私有仓库,我们就需要将镜像保存为文件上传到服务器上,载入镜像即可使用. 1.保 ...
最新文章
- Java入门之初识设计模式---单列模式
- 19.12 添加自定义监控项目;19.13,19.14 配置邮件告警(上下);19.15 测试告警19.16...
- wms仓储系统培训_WMS系统开创智慧仓储新方向
- pop to 特定的UIViewController
- 使用 Syslog 连接外部解决方案
- 【体系结构】Oracle段区块的个人理解
- String 和 StringBuffer 的区别
- redux-actions入门
- 系统分析员备考之经济管理篇(二)
- 【python之旅】python简介和入门
- Python中如何输入一个整数实例
- Cause: java.lang.ArrayIndexOutOfBoundsException: 8
- 艾米丽的蜜月旅行Android,美味餐厅:艾米丽的蜜月旅行完整版
- 暂时性死区(TDZ)
- 秘宝 联想正式进入元宇宙 互联网巨头的数字藏品用的哪种技术?
- 复古传奇服务器维护时间,复古传奇手游刷怪时间
- CSS小游戏2048—简单小游戏编程有手就行~会玩?你会写吗?
- 傅里叶级数FS,连续时间傅里叶变换CTFT,离散时间傅里叶变换DTFT,离散傅里叶变换DFT,推导与联系(一)
- 用POWELL法求极小值:Rosenbrock函数
- win10安装软件时显示:“现在更新设备, 若要继续接收应用建议,请立即更新”