在Caffe中,当您在protobuf文件中定义NN的输入时,您可以输入"数据"和"标签"。我猜测标签包含训练数据的预期输出(通常被认为是机器学习文献中的y
值)。
我的问题是在caffe.proto文件中,label被定义为标量(int或long)。至少对于数据,我可以将它设置为numpy数组,因为它需要String值。如果我正在训练多个预测输出,我怎么能把它作为数组传递?
或者我错了?什么是label
?它是为了什么?我怎样才能将y
值传递给caffe?
答案 0 :(得分:3)
caffe的基本用例曾经是图像分类:为每个输入图像分配一个整数标签。因此,"数据"数据结构为4D浮点数组(3个通道图像的批量)和一个整数"标签"保留空间。批量中的每个图像。
使用HDF5输入数据层可以轻松克服此限制 例如,参见this answer。