python - 为张量流分割模型获取混淆矩阵的有效方法是什么？

Question

我有一个张量流分割模型（u-net），我在具有 4 个分割类的图像上运行。（我也有一个 2 类模型，但想法是一样的。）我想建立一个混淆矩阵来比较结果图像的类分配。

通过在图像上运行模型并使用 numpy 操作来构建相应的矩阵，我可以做到这一点。但是，我确信在 tensorflow 中必须有某种方法可以做到这一点。我正在寻找使用评估生成器的方法，这是我目前正在做的以获得准确性，但任何等效的替代方案都将受到欢迎。

标签： pythontensorflow