使用caffe中compute_image_mean计算数据集均值

1. 计算均值

在进行深度学习网络训练的时候,为了提升训练的精度等,需要数据集的均值,那么该均值文件是怎么来的呢?这是通过安装caffe完成之后tools目录下的compute_image_mean实现的。调用它去计算数据集的均值,再在训练的时候调用它就好了,下面是调用它实现数据集均值计算的例子。

./compute_image_mean ~/Desktop/handwrite_chinese/train_data/ ~/Desktop/handwrite_chinese/handwrite_mean.binaryproto

参数解释:
~/Desktop/handwrite_chinese/train_data/:是存放训练lmdb二进制文件的目录,在之前的博客中讲过该文件是如何生成的
~/Desktop/handwrite_chinese/handwrite_mean.binaryproto:是保存生成之后的均值文件路径

2. 使用方法

这里直接使用python的接口将其写在了网络定义中

data, label = L.Data(source=lmdb_path, backend=P.Data.LMDB, batch_size=batch_size,
    ntop=2, transform_param=dict(scale=0.00390625, mean_file='./handwrite_mean.binaryproto'))

使用之后的网络定义为:

layer {
  name: "Data1"
  type: "Data"
  top: "Data1"
  top: "Data2"
  transform_param {
    scale: 0.00390625
    mean_file: "./handwrite_mean.binaryproto"
  }
  data_param {
    source: "val_data"
    batch_size: 64
    backend: LMDB
  }
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值