Created
August 27, 2016 14:24
-
-
Save dalgu90/7fd265e5419efe71477ca560801d0130 to your computer and use it in GitHub Desktop.
convert from proto mean file to npy file
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import sys | |
CAFFE_ROOT = '../../' | |
sys.path.insert(0, CAFFE_ROOT + 'python/') | |
import caffe | |
import numpy as np | |
if len(sys.argv) != 3: | |
print "Usage: python protomean_to_npy.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 ) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thank you! Kaiming He!!