imagenet-vgg-verydeep-19参数解析

最近学习风格迁移用到vgg19,这里简要总结要迁移学习vgg的模型结构
下载链接http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat

参数结构

import scipy.io
import scipy.misc
path = "/Users/enjlife/deep-learning-model/imagenet-vgg-verydeep-19.mat"
vgg = scipy.io.loadmat(path)
vgg.keys()

查看vgg字典,发现共有6个key
dict_keys(['__header__', '__version__', '__globals__', 'layers', 'classes', 'normalization'])
这里我们关注layers key,里面是我们需要的参数
vgg['layers'].shape获取shape结果为(1, 43),正好与我们的模型层数相符合。
在这里插入图片描述
下面取第一层layer查看vgg['layers'][0][0].shape,结果显示是一个二维(1, 1)数组。
查看其元素内容发现vgg['layers'][0][0]包含参数部分和说明部分dtype=[('weights', 'O'), ('pad', 'O'), ('type', 'O'), ('name', 'O'), ('stride', 'O')],上述为说明部分,参数太长这里没有贴。
vgg['layers'][0][0][0][0][0]最后一个零选取的是W和b的参数。
vgg['layers'][0][0][0][0][1]最后一个1选组的是pad填充的参数。依次可以取出layer的类型、名字、步长。
这里还要注意,取出的W和b的矩阵依然是一个二维矩阵vgg['layers'][0][0][0][0][0].shape,结果为(1,2),然后我们通过vgg['layers'][0][0][0][0][0][0][0]vgg['layers'][0][0][0][0][0][0][1]取到W和b的值。

引用一张图片来总结下结果
在这里插入图片描述
imagenet-vgg-verydeep-19参数结构解析

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值