convert_to_imagenet_mean_npy.py

# -*- coding: utf-8 -*-

import os
import sys
import numpy as np
import caffe

if __name__ == "__main__":
    data = open("imagenet_mean.binaryproto", "rb").read()

    blob = caffe.proto.caffe_pb2.BlobProto()
    blob.ParseFromString(data)

    arr = np.asarray(caffe.io.blobproto_to_array(blob))
    print arr.shape

    np.save("imagenet_mean.npy", arr)