image-processing - 什么是像素级 softmax 损失？

Question

什么是像素级 softmax 损失？在我的理解中，这只是一个交叉熵损失，但我没有找到公式。有人能帮我吗？最好有pytorch代码。

标签： image-processingcomputer-visionpytorchimage-segmentationloss

Answer 1

您可以在此处阅读所有相关信息（那里还有指向源代码的链接）。

正如您已经观察到的，“softmax 损失”基本上是一种交叉熵损失，其计算结合了 softmax 函数和数值稳定性和效率的损失。
在您的示例中，损失是针对逐像素预测计算的，因此您有逐像素预测、逐像素目标和逐像素损失项。