如何讀取標籤數據，如果在朱古力

Question

從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結構中看到標籤關鍵字。

Answer 1

我總是用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查看詳細

培訓網絡