什么是"标签"在Caffe?

时间:2015-10-27 23:11:20

标签: machine-learning neural-network deep-learning caffe pycaffe

在Caffe中,当您在protobuf文件中定义NN的输入时,您可以输入"数据"和"标签"。我猜测标签包含训练数据的预期输出(通常被认为是机器学习文献中的y值)。

我的问题是在caffe.proto文件中,label被定义为标量(int或long)。至少对于数据,我可以将它设置为numpy数组,因为它需要String值。如果我正在训练多个预测输出,我怎么能把它作为数组传递?

或者我错了?什么是label?它是为了什么?我怎样才能将y值传递给caffe?

1 个答案:

答案 0 :(得分:3)

caffe的基本用例曾经是图像分类:为每个输入图像分配一个整数标签。因此,"数据"数据结构为4D浮点数组(3个通道图像的批量)和一个整数"标签"保留空间。批量中的每个图像。

使用HDF5输入数据层可以轻松克服此限制 例如,参见this answer