1

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

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

还是我弄错了?是什么label?它是干什么用的?以及如何将y值传递给 caffe?

4

1 回答 1

3

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

使用 HDF5 输入数据层可以轻松克服此限制。
参见例如这个答案

于 2015-10-28T06:39:16.523 回答