We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent befaf65 commit eabc503Copy full SHA for eabc503
src/deepsparse/loggers/metric_functions/cv/built_ins.py
@@ -136,11 +136,9 @@ def count_classes_detected(
136
and the values are their counts across the batch
137
"""
138
_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)
+ counter = Counter()
+ for detections in classes:
+ counter.update(detections)
144
# convert keys to strings if required
145
counter = {str(class_label): count for class_label, count in counter.items()}
146
return counter
0 commit comments