在 caffe 中使用 hdf5 中的平均文件答案

【问题标题】：Use mean file in hdf5 in caffe在 caffe 中使用 hdf5 中的平均文件
【发布时间】：2017-01-17 14:51:50
【问题描述】：

我正准备使用 hdf5 文件中的数据在 Caffe 中进行训练。该文件还包含训练集的每像素平均数据/图像。在'train_val.prototxt' 部分中的输入数据层文件'transform_params' 中，可以使用mean_file 来规范化数据，通常采用二进制原型格式，例如ImageNet Caffe tutorial example：

transform_param {
  mirror: true
  crop_size: 227
  mean_file: "data/ilsvrc12/imagenet_mean.binaryproto"
}

对于每个通道的标准化，可以使用 mean_value 代替 mean_file。

但是有没有办法直接从我的数据库（这里是 hdf5）文件中使用平均图像数据？

我已经从 hdf5 提取平均值到一个 numpy 文件，但不确定它是否可以在 prototxt 中使用或转换。我在 Caffe 文档中找不到这方面的信息。

【问题讨论】：

标签： machine-learning computer-vision neural-network deep-learning caffe

【解决方案1】：

AFAIK，"HDF5Data" 层不支持转换。当您将数据存储到 HDF5 文件时，您应该自己减去平均值。

如果你想以二进制proto格式保存一个numpy数组，可以查看this answer了解更多详情。

【讨论】：

感谢您的回答！是的，似乎是这样。写完这个问题后，我从 Caffe 得到了这个：“HDF5Data 不会转换数据。”