Caffe的数据层如下图所示:
其中的mean_file是求得训练图像的均值,它的计算可以直接使用命令行来实现,如:
#第一个参数是Caffe自带的计算均值的可执行文件
#第二个参数是Caffe用于训练的lmdb数据源mnist_train_lmdb (前面博客中有讲过怎样说生成Caffe的lmdb数据源)
#第三个参数是生成的mean.binaryproto文件存放的位置
#在ubuntu系统的命令行中输入:·
sudo /home/pcb/caffe/build/tools/compute_image_mean /home/pcb/caffe/examples/mnist/mnist_train_lmdb /home/pcb/caffe/examples/mnist/mean.binaryproto
最终在文件夹/home/pcb/caffe/examples/mnist下就会生成一个mean.binaryproto文件!