【发布时间】：2019-12-03 13:46:17
【问题描述】：
什么是像素级 softmax 损失？在我的理解中，这只是一个交叉熵损失，但我没有找到公式。有人能帮我吗？最好有pytorch代码。
【问题讨论】：
标签： image-processing computer-vision pytorch image-segmentation loss
【发布时间】：2019-12-03 13:46:17
【问题描述】：
您可以阅读here 的全部内容（那里还有指向源代码的链接）。
正如您已经观察到的，“softmax 损失”基本上是一种交叉熵损失，其计算结合了 softmax 函数和损失，以实现数值稳定性和效率。
在您的示例中，损失是针对逐像素预测计算的，因此您拥有逐像素预测、逐像素目标和逐像素损失项。
【讨论】：
-
所以它只是一个交叉熵？我得到了一个形状为 [batch, 8, h, w] 的张量，我的目标是 [batch, 1, h, w]，目标值的范围是 1 到 8。所以我可以计算像素级的 softmax 损失直接使用公式cross-entropy(got, target)?