probabilities for each class #9117
Answered
by
ZwwWayne
ArkhamKnight-ZZZ
asked this question in
Q&A
-
I used faster-rcnn network for target detection, how can I obtain the probabilities distribution for each class? |
Beta Was this translation helpful? Give feedback.
Answered by
ZwwWayne
Oct 23, 2022
Replies: 1 comment
-
The output of the model already contains the information. Its format is List[List[boxes, scores]]], where the outer list is for classes and the inner list indicates the number of boxes of that class. The box has 5 dimension where the 5th dimension is the score. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
ZwwWayne
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
The output of the model already contains the information. Its format is List[List[boxes, scores]]], where the outer list is for classes and the inner list indicates the number of boxes of that class. The box has 5 dimension where the 5th dimension is the score.