I guess that my question is: how do you apply BALD to a semantic segmentation problem? I guess the project I posted above refers to a classification problem where he applies softmax to get one single number with the classes probability and then calculates entropy based on that.
In the case of semantic segmentation, we have a pixel wise probability. How can you calculate the entropy from that?