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 caffe import numpy as np import sys if 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 )