From e01e495c92d05c4b48ef240956d3f47cab0b0163 Mon Sep 17 00:00:00 2001 From: kirilllzaitsev Date: Thu, 6 Jul 2023 22:27:53 +0200 Subject: [PATCH] rm test setup from plot() --- supervision/metrics/detection.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/supervision/metrics/detection.py b/supervision/metrics/detection.py index dd482643..4f660d22 100644 --- a/supervision/metrics/detection.py +++ b/supervision/metrics/detection.py @@ -221,17 +221,7 @@ class ConfusionMatrix: array = self.matrix.copy() - # num_classes = self.num_classes + 1 - num_classes = 21 - class_names = class_names[:20] - array = array[:20, :20] - array[0, 0] = 10 - array[1, 8] = 15 - array[2, 9] = 20 - array[3, 10] = 25 - array[4, 11] = 10 - array[5, 12] = 15 - array[8, 12] = 35 + num_classes = self.num_classes + 1 if normalize: eps = 1e-8