-
Python类object_detection_confusion_matrix,其功能为计算对象检测任务的混淆矩阵
资源介绍
用于目标检测的混淆矩阵
ConfusionMatrix类可用于为对象检测任务生成混淆矩阵。
用法
在测试代码中,您需要使用适当的参数声明ConfusionMatrix类。
conf_mat = ConfusionMatrix(num_classes = 3, CONF_THRESHOLD = 0.3, IOU_THRESHOLD = 0.5)
该类具有一个名为process_batch的函数,您可以使用它来更新混淆矩阵。
用法示例可以是:
conf_mat.process_batch(preds, gt_boxes)其中preds是模型做出的预测,[N,6] x1,y1,x2,y2,置信度,类和gt_boxes是地面真相标签,[M,4] x1,y1,x2 ,y2,班级。
参考
该存储库使用Pytorch存储库中的函数
和
代码与下面的存储库非常相似,该存储库的主要贡献是Co