您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

使用caffe训练CSV数据的Lenet

使用caffe训练CSV数据的Lenet

我假设您有一个hdf5数据文件'data/mnist_train_h5.hd5'

错误消息中可以看到,"HDF5Data"图层不支持数据转换。具体来说,您无法按图层缩放数据。 因此,您希望进行的任何转换都必须在创建时 自己 应用'data/mnist_train_h5.hd5'

"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
  }
}
dotnet 2022/1/1 18:50:14 有488人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶