1. 程式人生 > >caffe Python API 之中值轉換

caffe Python API 之中值轉換

# 編寫一個函式,將二進位制的均值轉換為python的均值
def convert_mean(binMean,npyMean):
    blob = caffe.proto.caffe_pb2.BlobProto()
    bin_mean = open(binMean, 'rb' ).read()
    blob.ParseFromString(bin_mean)
    arr = np.array( caffe.io.blobproto_to_array(blob) )
    npy_mean = arr[0]
    np.save(npyMean, npy_mean )

# 呼叫函式轉換均值
binMean='examples/cifar10/mean.binaryproto' npyMean='examples/cifar10/mean.npy' convert_mean(binMean,npyMean)