binaryproto转成.npy

来源:互联网 发布:网络翻译兼职靠谱吗 编辑:程序博客网 时间:2024/06/05 15:23

https://www.zhihu.com/question/28102651/answer/39397649


io模块里面的blobproto_to_array,能够生成npy


用下面的代码在需要转换均值文件的目录下构建convert_protomean.py文件,然后调用

python convert_protomean.py mean.binaryproto cifar10_mean.npy

import caffeimport numpy as npimport sysif len(sys.argv) != 3:print "Usage: python convert_protomean.py proto.mean out.npy"sys.exit()blob = caffe.proto.caffe_pb2.BlobProto()data = open( sys.argv[1] , 'rb' ).read()blob.ParseFromString(data)arr = np.array( caffe.io.blobproto_to_array(blob) )out = arr[0]np.save( sys.argv[2] , out )

0 0
原创粉丝点击