【Python】【Caffe】三、生成.npy均值文件《python调用caffe模块》

GitHub代码地址：https://github.com/HandsomeHans/Use-Python-to-call-Caffe-module

前言

我们一般都会用caffe提供的make_mean.sh生成prototxt格式的均值文件，并且这个脚本最后也会在终端输出三个均值。

下面代码就是如果将prototxt或者均值转换成.npy格式的均值文件。

一、prototxt转换为npy

import caffeimport numpy as npprototxt = 'doc/mean.binaryproto'npy = 'doc/mean.npy'blob = caffe.proto.caffe_pb2.BlobProto()data = open(prototxt, 'rb' ).read()blob.ParseFromString(data)array = np.array(caffe.io.blobproto_to_array(blob))mean_npy = array[0]np.save(npy ,mean_npy)

二、均值转换为npy

import numpy as npnpy = 'doc/mean.npy'mean = np.ones([3,256, 256], dtype=np.float) #256是图像尺寸mean[0,:,:] = 100mean[1,:,:] = 110mean[2,:,:] = 120np.save(npy, mean)

以上内容参考自：http://blog.csdn.net/hyman_yx/article/details/51732656