#一.caffe的python接口下,数据层的定义,均值文件的转化
- 数据格式的设置(以测试caffeNet为例)
import caffe
import sys
sys.path.append('.')
sys.path.append('/home/***/caffe') #根据自己的caffe路径修改,下面两行也一样
sys.path.append('/home/***/caffe/python')
sys.path.append('/home/***/caffe/python/caffe')
import numpy as np
model_file = './models/caffenet_train_iter_40000.caffemodel' #权重文件
mean_file = './models/caffeNet_mean.npy' #均值文件位.npy格式
deploy_file = './models/deploy.prototxt' #网络文件
caffe.set_mode_gpu() #设置为gpu模式,如果需要cpu运行,改为caffe.set_mode_cpu()
net = caffe.Net(deploy_file, model_file, caffe.TEST) #初始化网络
# 对输入数据进行变换
# caffe.io.transformer是一个类,实体化的时候构造函数__init__(self, inputs)给一个初值
transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})
transformer.set_mean('data', np.load(mean_file).mean(1).mean(1)) #加载均值文件
transformer.set_transpose('data', (2, 0, 1)) #由于python存储图片的格式位H*W*Channels,需要转化位 Channel*W,H
transformer.set_raw_scale('data', 255) #由于python保存图像数据是0-1,所以要转化位0-255
transformer.set_channel_swap('data', (2, 1, 0)) #将BGR转化位RGB
net.blobs['data'].reshape(1, 3, 227, 227) #设置输入数据的格式
img = caffe.io.load_image(imgPath) #读取图片,imgPath是图片路径
net.blobs['data'].data[...] = transformer.preprocess('data', img) #加载数据
net.forward() #前向传播
predicts = net.blos['prob'][0] #读取caffeNet最后一层的数据
- 转化均值文件
- 由于C++模式下均值文件为binaryproto格式,Python要求npy格式,转化代码
import caffe
import sys
sys.path.append('.')
sys.path.append('/home/***/caffe') #根据自己的caffe路径修改,下面两行也一样
sys.path.append('/home/***/caffe/python')
sys.path.append('/home/***/caffe/python/caffe')
import numpy as np
MEAN_PROTO_PATH = './../mean.binaryproto' #已存在均值文件路径
MEAN_NPY_PATH = './../caffeNet_mean.npy' #生成均值文件路径
blob = caffe.proto.caffe_pb2.BlobProto() # 创建protobuf blob
data = open(MEAN_PROTO_PATH, 'rb' ).read() #读取均值文件
blob.ParseFromString(data) #解析文件内容到blob
array = np.array(caffe.io.blobproto_to_array(blob)) #将blob中的均值转换成numpy格式,array的shape (mean_number,channel, hight, width)
mean_npy = array[0] #一个array中可以有多组均值存在,需要通过下标选择其中一组均值
np.save(MEAN_NPY_PATH ,mean_npy) #保存