I’m using detectron2 maskRCNN for semantic segmentation. The standard output of model prediction only includes the confidence of predicted class. Since I’m going to smooth classification across time, it will be helpful to know the confidence/probability of other classes. I wonder if this is feasible.