caffe使用均值檔案生成.npy檔案的Python程式碼,供後面模型測試使用
阿新 • • 發佈：2018-12-15
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)
