首先要去下载何恺明大神的Caffee权重文件,然后使用Chainer提供的功能直接转换。
按照网页https://github.com/KaimingHe/deep-residual-networks指示,需要去onedrive下载.
下载完毕后转换一下就好了。
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