文件名称：object_detection_confusion_matrix:Python类，用于计算对象检测任务的混淆矩阵
文件大小：3KB
文件格式：ZIP
更新时间：2024-05-30 18:15:17
deep-learning object-detection confusion-matrix object-detection-metrics Python
用于目标检测的混淆矩阵 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
【文件预览】：
object_detection_confusion_matrix-master
----LICENSE(1KB)
----README.md(1KB)
----confusion_matrix.py(4KB)