2016-09-24 30 views
0

从MemoryData读取数据这是net.prototxt我的数据层:如何读取标签数据,如果在朱古力

layer { 
    name: "csv" 
    type: "MemoryData" 
    top: "data" 
    top: "label" 
    include { 
     phase: TRAIN 
    } 
    memory_data_param { 
     batch_size: 10 
     channels: 1 
     width: 14 
     height: 1 
    } 
} 

我找到函数

MemoryDataLayer<Dtype>::Reset(Dtype* data, Dtype* labels, int n) 

,但我不知道在哪里我应该添加此功能?

现在我想知道 是哪里的标签数据?因为我只在Datum结构中看到标签关键字。

回答

0

我总是用MemoryData层,当我通过pycaffe module.Like这个

solver = caffe.SGDSolver(solver_file) 

X = np.zeros((batch_size, 3, im_height, im_width), dtype = np.float32) 
Y = np.zeros((batch_size,), dtype = np.float32) 
# put processed images into X, put labels into Y 

solver.net.set_input_arrays(X,Y) 

可以参考caffe_root /蟒蛇/朱古力/ pycaffe.py和_caffe.cpp查看详细

培训网络