TensorFlow 预训练模型导入

TensorFlow 预训练模型导入

模型下载地址

Pre-trained ModelsIndex of /models/tf/给出了大部分较新的、性能较好的网络预训练模型,包括VGG、ResNet、各个版本的Inception等。
要测试AlexNet模型,可以到AlexNet implementation + weights in TensorFlow下载相应的.npy文件,并按照提供的.py文件重建网络,注意该网络需要特征分组训练,并且输入像素值范围为 [0,255] [ 0 , 255 ] 。根据提供者的描述,该模型实际上由caffemodel转化而来。

【注】tensorflow是不能直接导入caffemodel的,需要将caffemodel转成tensorflow可以读取的格式,具体参见GitHub项目caffe-tensorflow

方法

对于不同后缀名的模型,需要不同的加载方式

  1. .npy文件、.npz文件、.mat文件
    这些文件保存的仅仅是网络的权重,因此为了重建网络,需要根据网络结构一步步搭建,将相应的权重赋予其中的 W,b

  • 2
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值