在 caffe 中使用 hdf5 中的均值文件

Question

我正准备使用 hdf5 文件中的数据在 Caffe 中进行训练。该文件还包含训练集的每像素均值 data/image。在 'transform_params' 部分输入数据层的文件 'train_val.prototxt' 中，可以使用 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 文档中找不到关于此的信息。

Answer 1

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

如果你想将一个numpy数组保存为binaryproto格式，你可以查看this answer了解更多详情

在 caffe 中使用 hdf5 中的均值文件

Use mean file in hdf5 in caffe

machine-learning

computer-vision

neural-network

deep-learning

caffe