图片在减均值后,会提高训练与预测的精度和速度。

1、二进制格式的均值计算

../caffe/build/tools/Release/compute_image_mean ../file_name_lmdb /../mean.binaryproto

具体命令如上。

2、Python格式的均值计算

需要用到Python接口或者进行特征可视化,需要Python格式的均值文件。 具体Python脚本如下:

import numpy as np
import sys,caffe

if len(sys.argv)!=3:
    print "Usage: python convert_mean.py mean.binaryproto mean.npy"
    sys.exit()

blob = caffe.proto.caffe_pb2.BlobProto()
bin_mean = open( sys.argv[1] , 'rb' ).read()
blob.ParseFromString(bin_mean)
arr = np.array( caffe.io.blobproto_to_array(blob) )
npy_mean = arr[0]
np.save( sys.argv[2] , npy_mean )

两个参数: 1、mean.binaryproto 二进制均值文件保存路径和文件名 2、mean.npy Python格式的均值文件保存路径和文件名