为 RetinaNet 对象检测模型生成混淆矩阵

问题描述 投票:0回答:1

有没有一种方法可以使用相同的测试集及其注释自动生成混淆矩阵?

这是RetinaNet模型

的链接

我尝试从图像分类模型中调整混淆矩阵,但这显然不起作用,我目前正在尝试使用运行 inference.py 和 dataset.py 的结果,因为第一个产生预测结果,第二个显示一些结果用于训练的数据集示例。

python pytorch object-detection confusion-matrix retinanet
1个回答
0
投票

您可以尝试使用 scikit lib,特别是:

Scikit 混淆矩阵

对于多标签问题,您可以使用:

Scikit 多标签混淆矩阵

© www.soinside.com 2019 - 2024. All rights reserved.