In their paper, the authors state that they calculate the energy function by a pixel-wise softmax over the final feature map in combination with the cross entropy loss funct