mmpretrain icon indicating copy to clipboard operation
mmpretrain copied to clipboard

multi label classification tasks

Open Warrior996 opened this issue 2 years ago • 2 comments

Is the MMClassification suitable for multi label classification tasks? eg. ChestX-ray14 dataset

Warrior996 avatar May 10 '22 14:05 Warrior996

Yes, we support multi-label tasks, You can implement your dataset by referring to the VOC dataset. And the head of model should be MultiLabelLinearHead.

mzr1996 avatar May 10 '22 14:05 mzr1996

@mzr1996 你好,我现在尝试对照VOC数据进行多标签任务,但是好像标签有问题,不知道可以不可以给一个较为完善的示例,非常感谢

CompliceU avatar Jun 10 '22 01:06 CompliceU

@CompliceU 目前以支持 VOC 数据集,可以直接运行config, 参考CSRA

Ezra-Yu avatar Sep 08 '22 02:09 Ezra-Yu

请问如何用CSRA训练自己的数据集呢,我制作成VOC架构的数据集格式 会出现 精度为0的现象

st929 avatar Dec 01 '23 11:12 st929