caffe with multi-label images

前端 未结 3 1415
忘掉有多难
忘掉有多难 2020-12-08 11:25

I have a dataset of images that have multiple labels; There are 100 classes in the dataset, and each image has 1 to 5 labels associated with them.

I\'m following the

3条回答
  •  无人及你
    2020-12-08 11:48

    caffe supports multilabel. You can put the labels into n-hot vectors e.g. [0,1,1,0,0,1,...] . You need to reshape the labels to n*k*1*1 tensors and use sigmoid cross-entropy or euclidean, not softmax (which forces sum(outputs)=1 )

提交回复
热议问题