classification-pytorch icon indicating copy to clipboard operation
classification-pytorch copied to clipboard

你好,这个任务修改之后可以完成多标签多类别 分类的任务吗

Open praymich opened this issue 1 year ago • 4 comments

如题

praymich avatar Jun 28 '23 02:06 praymich

额0 0你的意思是同时输出多个标签?

bubbliiiing avatar Jul 02 '23 13:07 bubbliiiing

sigmoid?

bubbliiiing avatar Jul 02 '23 13:07 bubbliiiing

额0 0你的意思是同时输出多个标签?

比如同时输出一张猫图像为“ 猫 ”和 “动物”,是不是可以采用类似下面的方法 `out_1 = self.fc_1(x)

out_2 = self.fc_2(x)

return out_1, out_2 在模型训练时,计算多个分类损失相加loss = criterion(pred1, y1) + criterion(pred2,y2)`

DeepSpace98 avatar Jul 13 '23 02:07 DeepSpace98

额0 0你的意思是同时输出多个标签?

比如同时输出一张猫图像为“ 猫 ”和 “动物”,是不是可以采用类似下面的方法 `out_1 = self.fc_1(x)

out_2 = self.fc_2(x)

return out_1, out_2 在模型训练时,计算多个分类损失相加loss = criterion(pred1, y1) + criterion(pred2,y2)`

我现在确实是这样弄的

praymich avatar Jul 13 '23 02:07 praymich