From 3d6ea57ad467fa1fca13448f1d6a9b3624f97ac8 Mon Sep 17 00:00:00 2001 From: "Kwabena W. Agyeman" Date: Thu, 15 May 2025 15:27:20 -0700 Subject: [PATCH] scripts/libraries: Remove tuple detection for np.nonzero. np.nonzero always outputs a tuple. --- scripts/libraries/ml/ml/postprocessing.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/scripts/libraries/ml/ml/postprocessing.py b/scripts/libraries/ml/ml/postprocessing.py index c00402486..ba8fcffde 100644 --- a/scripts/libraries/ml/ml/postprocessing.py +++ b/scripts/libraries/ml/ml/postprocessing.py @@ -105,9 +105,7 @@ class yolo_v2_postprocess: # Threshold all the scores score_indices = sigmoid(row_outputs[:, _YOLO_V2_SCORE]) - score_indices = np.nonzero(score_indices > self.threshold) - if isinstance(score_indices, tuple): - score_indices = score_indices[0] + score_indices = np.nonzero(score_indices > self.threshold)[0] if not len(score_indices): return _NO_DETECTION @@ -173,9 +171,7 @@ class yolo_v5_postprocess: # Threshold all the scores score_indices = row_outputs[:, _YOLO_V5_SCORE] - score_indices = np.nonzero(score_indices > self.threshold) - if isinstance(score_indices, tuple): - score_indices = score_indices[0] + score_indices = np.nonzero(score_indices > self.threshold)[0] if not len(score_indices): return _NO_DETECTION