search

kaanakan / object_detection_confusion_matrix

Python class for calculating confusion matrix for object detection task
MIT License
85 stars 18 forks source link

questions on parameter:num_classes #9

Closed zhangxiwensjtu closed 3 years ago

zhangxiwensjtu commented 3 years ago

hi ,thanks for the code. but I am confused about the parameter num_classes. I find that in init function , matrix's size is (num_classes+1, num_classes+1), what does +1 refer to? I now have a detection task for 19 classes ,should i let num_classes equal to 19 or 18?

zhangxiwensjtu commented 3 years ago

sorry to interrupt. From other guy's issue, I understand it . For 19 classes' detection, the 20th row/col means background FN or FP, I'll close this issue myself, hope this will help someone who has the same question as I have.