0

I was wondering if it's possible to predict a 1-by-n feature associated to an input image using caffe. In this post there is a solution to make caffe predict a binary vector such as [1 0 1 0].

Is this solution also suitable if I have a 1-by-n float vector as label (such as [0.2, 0.1, 0.3, 0.4] ? I want to predict such a vector, not a binary vector label.

Community
  • 1
  • 1
user2614596
  • 630
  • 2
  • 11
  • 30

2 Answers2

1

You can also think about this MultiTaskData Layer. It can parse float typed label vector as you mentioned in your question.

Community
  • 1
  • 1
Dale
  • 1,608
  • 1
  • 9
  • 26
0

Yes; I believe that if you simply follow the instructions in the post you referenced, but use float labels in place of the binary ones, that you'll get useful results. Caffe is happy to support you. :-)

Prune
  • 76,765
  • 14
  • 60
  • 81