Skip to content

Commit 2007809

Browse files
author
Berkay Ugur Senocak
committed
corrected error type
1 parent ec6fcf7 commit 2007809

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

confusion_matrix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def process_batch(self, detections, labels: np.ndarray):
5151

5252
try:
5353
detections = detections[detections[:, 4] > self.CONF_THRESHOLD]
54-
except TypeError:
54+
except IndexError or TypeError:
5555
# detections are empty, end of process
5656
for i, label in enumerate(labels):
5757
gt_class = gt_classes[i]

0 commit comments

Comments
 (0)