轉換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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章