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 return
top_plate, top_char_confidences = license_plates[0], confidences[0] 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 # Check if we have a previously detected plate for this ID
if id in self.detected_license_plates: if id in self.detected_license_plates:
prev_plate = self.detected_license_plates[id]["plate"] prev_plate = self.detected_license_plates[id]["plate"]
prev_char_confidences = self.detected_license_plates[id]["char_confidences"] prev_char_confidences = self.detected_license_plates[id]["char_confidences"]
prev_avg_confidence = sum(prev_char_confidences) / len( prev_avg_confidence = (
prev_char_confidences (sum(prev_char_confidences) / len(prev_char_confidences))
if prev_char_confidences
else 0
) )
# Define conditions for keeping the previous plate # Define conditions for keeping the previous plate