首先要去下載何愷明大神的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