我正在尝试将数据加载到网络中,因为我需要自定义数据输入(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 /形状,但我找不到任何培训。感谢。