Caffe：npy与binaryproto互相转换

1.binaryproto转npy

import numpy as np
import sys

caffe_root = 'path/caffe/'
sys.path.insert(0, caffe_root + 'python')
import caffe

blob = caffe.proto.caffe_pb2.BlobProto() 
data = open('ilsvrc_2012_mean.binaryproto', 'rb').read()
blob.ParseFromString(data)
arr = np.array(caffe.io.blobproto_to_array(blob)) 
out = arr[0]
np.save('ilsvrc_2012_mean.npy', out)

2.npy转binaryproto

import numpy
import sys

caffe_root = 'path/caffe/' 
sys.path.insert(0, caffe_root + 'python')
import caffe

blob = caffe.proto.caffe_pb2.BlobProto()
with open('ilsvrc_2012_mean.npy','rb') as f:
  data_mean = numpy.load(f)
blob.channels=1
blob.height = data_mean.shape[0]
blob.width = data_mean.shape[1]
blob.data.extend(data_mean.astype(float).flat)
binaryproto_file = open('ilsvrc_2012_mean.binaryproto', 'wb') 
binaryproto_file.write(blob.SerializeToString())
binaryproto_file.close()