Chi_Laidian的博客

原创 多标签分类损失函数与结果显示

菜鸟认知，大佬误喷网络输出格式为（num，num_class, h, w ）,标签为（num，h , w ），标签不进行独热编码，标签.unique =[0,1,2，3，···，num_class]，使用损失函数nn.CrossEntropyLoss()。预测结果显示以num=1为例per.shape为（1，num_class, h, w ）def plt_predict(per,n_class): per_ = torch.sigmoid(per) per_ = p