分类：nn.NLLLoss2d

2020
10-08

Pytorch损失函数nn.NLLLoss2d()用法说明

最近做显著星检测用到了NLL损失函数对于NLL函数，需要自己计算log和softmax的概率值，然后从才能作为输入输入[batch_size,channel,h,w]目标[batch_size,h,w]输入的目标矩阵，每个像素必须是类型.举个例子。第一个像素是0，代表着类别属于输入的第1个通道；第二个像素是0，代表着类别属于输入的第0个通道，以此类推。x=Variable(torch.Tensor([[[1,2,1],[2,2,1],[0,1,1]],[[0,1,3],[... 继续阅读 >