Как рассчитать кросс-энтропию в 3D-изображении?

Question

См. Рисунок здесь

осталось значение (2, 480, 640) и его значение softmax *

значение равно (2, 480, 640) иэто однозначное значение кодирования

как получить кросс-энтропийную потерю во всех элементах?

Answer 1

Точно так же, как с любым другим изображением.Используйте binary_cross_entropy (слева, справа) .Обратите внимание, что

1. Оба должны иметь torch.float32 dtype, поэтому вам может понадобиться сначала преобразовать right, используя right.to(torch.float32).
2. Если ваш тензор left содержитлогиты вместо вероятностей лучше назвать binary_cross_entropy_with_logits (слева, справа) , чем binary_cross_entropy(torch.sigmoid(left), right)