登录 注册
当前位置:主页 > 资源下载 > 45 > Python类object_detection_confusion_matrix,其功能为计算对象检测任务的混淆矩阵

Python类object_detection_confusion_matrix,其功能为计算对象检测任务的混淆矩阵

  • 更新:2024-08-06 16:26:08
  • 大小:3KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

用于目标检测的混淆矩阵 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