Where I can get probabilities distributions in the output?

[josematez]

Hello, I'm interested in this work as an extension of Detectron2 to include probabilities distributions instead of confidence scores of just the top-1 class. However, I'm running the demo.py code and trying some different CenterNet models, but I'm not able to extract probabilities distributions, just a single score of the top-1 class per bounding box. Could you help me? Thanks in advance!