search

kaanakan / object_detection_confusion_matrix

Python class for calculating confusion matrix for object detection task
MIT License
84 stars 17 forks source link

Input data formats #11

Open IamExperimenting opened 2 years ago

IamExperimenting commented 2 years ago

Hi @kaanakan, thanks for implementing this. Could you please post some sample ground truth and predictions data. So we can refer that produce the same.

tharindu326 commented 2 years ago 
det = [[349, 832, 470, 907, 0.9968715906143188, 0], [469, 789, 676, 886, 0.9854957461357117, 0], [759, 887, 907, 1065, 0.9964740872383118, 4], [1234, 736, 1341, 757, 0.7642012238502502, 11], [1683, 513, 1759, 556, 0.8026162385940552, 13], [1883, 438, 1919, 503, 0.7332593202590942, 13], [478, 293, 514, 338, 0.70100337266922, 13]]
gt = [[0, 575, 789, 677, 889], [0, 469, 806, 581, 860], [0, 349, 829, 472, 907], [1, 474, 289, 512, 342], [1, 1681, 511, 1755, 563], [1, 1873, 439, 1920, 506]]

inputs are like:

detections: np.array(det)
labels: np.array(gt)