使用caffe训练CSV数据的Lenet

我假设您有一个hdf5数据文件

'data/mnist_train_h5.hd5'

1. 从错误消息中可以看到，

   "HDF5Data"

   图层不支持数据转换。具体来说，您无法按图层缩放数据。
   因此，您希望进行的任何转换都必须在创建时 自己 应用

   'data/mnist_train_h5.hd5'

   。
   

2. "HDF5Data"

   层不接受

   data_param

   ，而是

   hdf5_data_param

   使用

   source

   指定hd5二进制文件 列表 的参数。在您的情况下，您应该准备一行额外的 文本 文件

   'data/mnist_train_h5.txt'

   ：

数据/mnist_train_h5.hd5

该文本文件将告诉caffe阅读

'data/mnist_train_h5.hd5'

结果层应如下所示：

layer {  name: "mnist"  type: "HDF5Data"  top: "data"  top: "label"  hdf5_data_param {    source: "data/mnist_train_h5.txt"    batch_size: 64  }  include {    phase: TRAIN  }}