jwyang / faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn
MIT License
7.7k stars 2.33k forks source link

How to obtain the confusion matrix? #910

Open yuanyulee34 opened 1 year ago

yuanyulee34 commented 1 year ago

How can I get the confusion matrix? I need to know the True Positive and False Positive values. I see detections.pkl in the output. I try to read the contents of the detections.pkl file, extract the real labels and prediction scores, and predict Scores converted to binary predictions, confusion matrix computed, but fails.