Skip to content

Commit eabc503

Browse files
Update src/deepsparse/loggers/metric_functions/cv/built_ins.py
Co-authored-by: Benjamin Fineran <[email protected]>
1 parent befaf65 commit eabc503

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/deepsparse/loggers/metric_functions/cv/built_ins.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -136,11 +136,9 @@ def count_classes_detected(
136136
and the values are their counts across the batch
137137
"""
138138
_check_valid_classes(classes)
139-
detected_classes = []
140-
for sample_idx, sample in enumerate(classes):
141-
if sample != [None]:
142-
detected_classes += sample
143-
counter = Counter(detected_classes)
139+
counter = Counter()
140+
for detections in classes:
141+
counter.update(detections)
144142
# convert keys to strings if required
145143
counter = {str(class_label): count for class_label, count in counter.items()}
146144
return counter

0 commit comments

Comments
 (0)