Open God-song opened 1 year ago
如何使用一个模型,分类多个任务呢,我目前使用PPLCNet_x1_0,分类四个方向,我除了想要分开四个方向,我还想分类,是类型的图片,我想要有两个输出头,第一个输出口输出4,代表4个方向,第二个输出口位图片类型,例如10中类型,我该怎么做呢
最简单的做法,就是把两个任务合并,改成4*10=40分类。 比较好的做法,就是需要改一下模型组网代码,将最后的一个fc,https://github.com/PaddlePaddle/PaddleClas/blob/release/2.5/ppcls/arch/backbone/legendary_models/pp_lcnet.py#L303 ,改成两个fc,分别输出4和10类,两个fc的结果可以concat到一起再return,然后loss需要重写,dataset也需要重写,支持多个label。
最简单的做法,就是把两个任务合并,改成4*10=40分类。 比较好的做法,就是需要改一下模型组网代码,将最后的一个fc, https: //github.com/PaddlePaddle /PaddleClas/blob/release/2.5/ppcls/arch/backbone/legendary_models/pp_lcnet.py#L303,改成两个fc,分别输出4和10类,两个fc的结果可以concat到一起再return,然后loss需要重写,数据集也需要重写,支持多个标签。 paddleclas不支持多任务吗
目前不支持多任务。
欢迎您使用PaddleClas并反馈相关问题,非常感谢您对PaddleClas的贡献! 提出issue时,辛苦您提供以下信息,方便我们快速定位问题并及时有效地解决您的问题: