Prevent division by zero in lpr confidence checks (#14615)

This commit is contained in:
Josh Hawkins 2024-10-26 17:27:02 -05:00 committed by Nicolas Mowen
parent 7c82fae009
commit e12b2dc68c

View File

@ -599,14 +599,20 @@ class EmbeddingMaintainer(threading.Thread):
return
top_plate, top_char_confidences = license_plates[0], confidences[0]
avg_confidence = sum(top_char_confidences) / len(top_char_confidences)
avg_confidence = (
(sum(top_char_confidences) / len(top_char_confidences))
if top_char_confidences
else 0
)
# Check if we have a previously detected plate for this ID
if id in self.detected_license_plates:
prev_plate = self.detected_license_plates[id]["plate"]
prev_char_confidences = self.detected_license_plates[id]["char_confidences"]
prev_avg_confidence = sum(prev_char_confidences) / len(
prev_char_confidences
prev_avg_confidence = (
(sum(prev_char_confidences) / len(prev_char_confidences))
if prev_char_confidences
else 0
)
# Define conditions for keeping the previous plate