administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-18 17:14:26 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix saving bug

Browse Source
This commit is contained in:
Weitheng Haw 2025-02-04 01:12:02 +00:00
parent 381e8d598d
commit c58d875a01
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/embeddings/lpr/lpr.py
View File

@ -195,11 +195,15 @@ class LicensePlateRecognition:
average_confidence = conf average_confidence = conf
# Save debug image # Save debug image
try:
save_image = cv2.cvtColor( save_image = cv2.cvtColor(
rotated_images[original_idx], cv2.COLOR_RGB2BGR rotated_images[original_idx], cv2.COLOR_RGB2BGR
) )
filename = f"{plate}_{int(average_confidence[0] * 100)}_{area}.jpg" confidence_value = average_confidence[0] if average_confidence and len(average_confidence) > 0 else 0
filename = f"{plate}_{int(confidence_value * 100)}_{area}.jpg"
cv2.imwrite(os.path.join(self.debug_dir, filename), save_image) cv2.imwrite(os.path.join(self.debug_dir, filename), save_image)
except Exception as e:
logger.warning(f"Failed to save debug image: {e}")
license_plates[original_idx] = plate license_plates[original_idx] = plate
average_confidences[original_idx] = average_confidence average_confidences[original_idx] = average_confidence