PaddleClas icon indicating copy to clipboard operation
PaddleClas copied to clipboard

多标签图像分类,topk参数设置不为1,预测结果仍是只有1个预测值

Open phoenixliu666 opened this issue 2 years ago • 1 comments

  1. PaddleClas版本:PaddleClas release/2.4
  2. image 继承父类的__call__方法,传入的multilabel参数为True会导致代码跳到else语句,结果也就是topk参数设置>1无效 image

phoenixliu666 avatar Aug 17 '22 15:08 phoenixliu666

hi,您好,多标签分类不应该有top-k这个约束,应该用阈值来卡结果,默认是0.5,你也可以尝试其他阈值,详情改动请见:https://github.com/PaddlePaddle/PaddleClas/pull/2219

cuicheng01 avatar Aug 22 '22 06:08 cuicheng01