为caffe创建带浮点数的hfd5

时间:2016-06-08 08:51:48

标签: neural-network regression deep-learning caffe multilabel-classification

我想提供标签的caffe矢量(多标签回归问题),所以我使用following link来创建hdf5文件。
使用此代码,我创建了一个image_list.txt,其中包含每行中文件和浮动标签的路径,例如。

/home/deep/00000.bmp  0.9997 0.0236 -0.0082 -0.0231 0.9980 0.0588 0.0096 -0.0586 0.9982 -0.0046 0.1084 0.3938

但它似乎只适用于整数标签 当我运行demo.m时出现以下错误:

  

使用dataread时出错       无法从文件读取整数(第1行,第3行)==> .9997 0.0236 -0.0082 -0.0231 0.9980   0.0588 0。

1 个答案:

答案 0 :(得分:0)

如果您还使用了caffe的C ++界面,我想与您分享Multitask DataLayer并且它支持用于回归任务的浮点型标签向量。 Here提供了一个学习面部表情的简单示例。标签分布,矢量大小为8。