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!
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!