mmpretrain
mmpretrain copied to clipboard
multi label classification tasks
Is the MMClassification suitable for multi label classification tasks? eg. ChestX-ray14 dataset
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 你好,我现在尝试对照VOC数据进行多标签任务,但是好像标签有问题,不知道可以不可以给一个较为完善的示例,非常感谢
@CompliceU 目前以支持 VOC 数据集,可以直接运行config, 参考CSRA
请问如何用CSRA训练自己的数据集呢,我制作成VOC架构的数据集格式 会出现 精度为0的现象