scripts/libraries: Remove tuple detection for np.nonzero.

np.nonzero always outputs a tuple.
This commit is contained in:
Kwabena W. Agyeman 2025-05-15 15:27:20 -07:00
parent 727d3200a2
commit 3d6ea57ad4

View File

@ -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