有没有一种方法可以使用相同的测试集及其注释自动生成混淆矩阵?
这是RetinaNet模型
我尝试从图像分类模型中调整混淆矩阵,但这显然不起作用,我目前正在尝试使用运行 inference.py 和 dataset.py 的结果,因为第一个产生预测结果,第二个显示一些结果用于训练的数据集示例。
您可以尝试使用 scikit lib,特别是:
Scikit 混淆矩阵
对于多标签问题,您可以使用:
Scikit 多标签混淆矩阵