使用Caffe变压器进行HDF5数据处理以进行培训

时间:2017-12-13 17:55:08

标签: python io caffe hdf5 transformer

我正在尝试将数据加载到网络中,因为我需要自定义数据输入(3个顶部:1个用于数据图像,2个用于不同标签)我使用HD5F文件加载数据。它看起来与此相似:

layer {
  name: "data"
  type: "HDF5Data"
  top: "img"
  top: "alabels"
  top: "blabels"
  include {
    phase: TRAIN
  }
  hdf5_data_param {
    source: "path_to_caffe/examples/hdf5_classification/data/train.txt"
    batch_size: 64
  }
} 

我想使用Caffe自己的Transformer(标准版)预处理图像,当我必须使用网络模型的数据blob初始化Transformer时,我该怎么做:

transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})

我发现的所有示例都首先加载一个预先训练好的网络,并设置一个变形数据,其数据blob /形状,但我找不到任何培训。感谢。

0 个答案:

没有答案