Create a binary proto file that contains a blob with the mean.

Create binary proto file with mean for Caffe

import sys
caffe_root = '/path/to/caffe/'
sys.path.insert(0, caffe_root + 'python')
import caffe.proto.caffe_pb2 as caffe_proto


def create_mean_file(filename, mean):
    """
    Create a binaryproto file with a single blob containing the given pixel-wise mean.
    Overwrites any existing file with the same name.
    :param filename: Filename to be used
    :type filename: str
    :param mean: Numpy array holding the mean
    :type mean: numpy.multiarray.ndarray
    """
    blob = caffe_proto.BlobProto()
    blob.shape.dim.extend(map(int, mean.shape))
    blob.data.extend(mean.flatten().tolist())
    # legacy
    blob.channels = mean.shape[0]
    blob.height = mean.shape[1]
    blob.width = mean.shape[2]
    blob.num = 1
    with open(filename, "wb") as f:
        f.write(blob.SerializeToString())