.binaryproto of caffe to numpy array

binaryproto2numpy.py

# Converts the mean.binaryproto (mean image) to a numpy image. 
import caffe
import numpy as np
import sys


blob = caffe.proto.caffe_pb2.BlobProto()
data = open('mean.binaryproto' , 'rb' ).read()
blob.ParseFromString(data)
arr = np.array( caffe.io.blobproto_to_array(blob) )
out = arr[0]
np.save( 'mean', out )
im = out.transpose(1,2,0)