转换Caffe模型权重文件到Chainer的npz格式

首先要去下载何恺明大神的Caffee权重文件,然后使用Chainer提供的功能直接转换。

按照网页https://github.com/KaimingHe/deep-residual-networks指示,需要去onedrive下载.

https://onedrive.live.com/?authkey=%21AAFW2-FVoxeVRck&id=4006CBB8476FF777%2117887&cid=4006CBB8476FF777

下载完毕后转换一下就好了。


from chainer.links.model.vision.resnet import ResNetLayers

'''
转换 caffee 模型权重文件到 Chainer npz 文件
Date 2019/08/28
'''

#ResNetLayers.convert_caffemodel_to_npz(path_caffemodel, path_npz, n_layers)
ResNetLayers.convert_caffemodel_to_npz("/home/henry/caffe-tensorflow/ResNet-101-model.caffemodel",
                                       "/home/henry/dataset/caffe_model/ResNet-101-model.npz",
                                       101)



源文件:

-rw-rw-r-- 1 henry henry 178662602 8月  27 21:22 ResNet-101-model.caffemodel

转换后:

-rw-rw-r-- 1 henry henry 178773064 8月  28 16:19 ResNet-101-model.npz

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值