caffe--- 图片数据均值计算小工具

图片减去均值后，再进行训练和测试，会提高速度和精度。因此，一般在各种模型中都会有这个操作。

caffe中使用的均值数据格式是binaryproto, 作者为我们提供了一个计算均值的文件

compute_image_mean.cpp，放在caffe根目录下的tools文件夹里面。

但是这要求先把图片转换为lmdb格式，如果是自己写的data层（一般用pyhon写），则需要另外写一个图片均值计算程序，下面是博主自己写的一个示例：

程序：

clc;

clear all;

close all;

%数据目录

base_line = 'C:\Users\10337\Desktop\PETA_F\data_8699\';

dirs_path = base_line;

dirs_list = dir([dirs_path, 'identity_*']);

img_mean=zeros(1,3);

sum_r=0;

sum_g=0;

sum_b=0;

image_size=224;

count=0;

for i=1:length(dirs_list)

    %子目录

    image_path= [dirs_path, dirs_list(i).name,'\'];

    image_list=dir([image_path,'*.jpg']);

    for j=1:length(image_list)

        count=count+1;

        image_name = [image_path, image_list(j).name];

        img=imread(image_name);

        img=imresize(img,[image_size,image_size],'bilinear');

        sum_r=sum_r+sum(sum(img(:,:,1)))/(image_size*image_size);

        sum_g=sum_g+sum(sum(img(:,:,2)))/(image_size*image_size);

        sum_b=sum_b+sum(sum(img(:,:,3)))/(image_size*image_size);

    end

end

sum_r=sum_r/count;

sum_g=sum_g/count;

sum_b=sum_b/count;

img_mean=[sum_r,sum_g,sum_b]   %  [114.4326  110.1417  106.5148]