diff --git a/ultralytics/utils/metrics.py b/ultralytics/utils/metrics.py index 2b5821cb..7d9629ec 100644 --- a/ultralytics/utils/metrics.py +++ b/ultralytics/utils/metrics.py @@ -440,7 +440,7 @@ class ConfusionMatrix: def print(self): """Print the confusion matrix to the console.""" - for i in range(self.nc + 1): + for i in range(self.matrix.shape[0]): LOGGER.info(" ".join(map(str, self.matrix[i])))