mirror of
https://github.com/openmv/openmv.git
synced 2025-09-26 23:09:13 +08:00
scripts/libraries: Remove tuple detection for np.nonzero.
np.nonzero always outputs a tuple.
This commit is contained in:
parent
727d3200a2
commit
3d6ea57ad4
@ -105,9 +105,7 @@ class yolo_v2_postprocess:
|
|||||||
|
|
||||||
# Threshold all the scores
|
# Threshold all the scores
|
||||||
score_indices = sigmoid(row_outputs[:, _YOLO_V2_SCORE])
|
score_indices = sigmoid(row_outputs[:, _YOLO_V2_SCORE])
|
||||||
score_indices = np.nonzero(score_indices > self.threshold)
|
score_indices = np.nonzero(score_indices > self.threshold)[0]
|
||||||
if isinstance(score_indices, tuple):
|
|
||||||
score_indices = score_indices[0]
|
|
||||||
if not len(score_indices):
|
if not len(score_indices):
|
||||||
return _NO_DETECTION
|
return _NO_DETECTION
|
||||||
|
|
||||||
@ -173,9 +171,7 @@ class yolo_v5_postprocess:
|
|||||||
|
|
||||||
# Threshold all the scores
|
# Threshold all the scores
|
||||||
score_indices = row_outputs[:, _YOLO_V5_SCORE]
|
score_indices = row_outputs[:, _YOLO_V5_SCORE]
|
||||||
score_indices = np.nonzero(score_indices > self.threshold)
|
score_indices = np.nonzero(score_indices > self.threshold)[0]
|
||||||
if isinstance(score_indices, tuple):
|
|
||||||
score_indices = score_indices[0]
|
|
||||||
if not len(score_indices):
|
if not len(score_indices):
|
||||||
return _NO_DETECTION
|
return _NO_DETECTION
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user