import sys,os
caffe_root ='/home/hadoop/caffe/'
sys.path.append('/home/hadoop/caffe/python')
import caffe
os.chdir(caffe_root)
import numpy as np
print "Usage: python convert_protomean.py proto.mean out.npy"
blob = caffe.proto.caffe_pb2.BlobProto()
BINARY_PROTO_FILE_NAME ='/home/hadoop/桌面/caffe_study/more_eyes/imagenet_mean.binaryproto'
BINARY_PROTO_FILE_PATH = os.path.join(os.getcwd(),BINARY_PROTO_FILE_NAME)
NPY_FILE_NAME ='/home/hadoop/桌面/caffe_study/more_eyes/mean.npy'
NPY_FILE_PATH = os.path.join(os.getcwd(),NPY_FILE_NAME)
data = open(BINARY_PROTO_FILE_PATH, 'rb' ).read()
blob.ParseFromString(data)
arr = np.array(caffe.io.blobproto_to_array(blob))
out = arr[0]
np.save(NPY_FILE_PATH,out)
caffe使用均值文件生成.npy文件的Python代码,供后面模型测试使用
最新推荐文章于 2020-01-04 14:39:53 发布