ahasera commited on
Commit
bf07f51
·
1 Parent(s): 19fabdc
Files changed (2) hide show
  1. app.py +13 -2
  2. models/yolov8n.pt +3 -0
app.py CHANGED
@@ -84,10 +84,21 @@ def read_license_plate(license_plate_crop, ocr_engine='easyocr'):
84
  detections_raw = [(None, text_raw.strip(), None)]
85
  detections_preprocessed = [(None, text_preprocessed.strip(), None)]
86
  elif ocr_engine == 'paddleocr':
 
87
  result_raw = ocr_paddle.ocr(license_plate_crop)
88
  result_preprocessed = ocr_paddle.ocr(preprocessed_image)
89
- detections_raw = [(None, result_raw[0][0][1][0], result_raw[0][0][1][1] if result_raw[0] else 0.0)]
90
- detections_preprocessed = [(None, result_preprocessed[0][0][1][0], result_preprocessed[0][0][1][1] if result_preprocessed[0] else 0.0)]
 
 
 
 
 
 
 
 
 
 
91
  else:
92
  raise ValueError(f"OCR engine '{ocr_engine}' not supported.")
93
 
 
84
  detections_raw = [(None, text_raw.strip(), None)]
85
  detections_preprocessed = [(None, text_preprocessed.strip(), None)]
86
  elif ocr_engine == 'paddleocr':
87
+ preprocessed_image = preprocess_image(license_plate_crop) # Assurez-vous que cette ligne est incluse
88
  result_raw = ocr_paddle.ocr(license_plate_crop)
89
  result_preprocessed = ocr_paddle.ocr(preprocessed_image)
90
+
91
+ # Vérifiez si les résultats ne sont pas vides avant de les utiliser
92
+ if result_raw and result_raw[0]:
93
+ detections_raw = [(None, result_raw[0][0][1][0], result_raw[0][0][1][1])]
94
+ else:
95
+ detections_raw = [(None, '', 0.0)]
96
+
97
+ if result_preprocessed and result_preprocessed[0]:
98
+ detections_preprocessed = [(None, result_preprocessed[0][0][1][0], result_preprocessed[0][0][1][1])]
99
+ else:
100
+ detections_preprocessed = [(None, '', 0.0)]
101
+
102
  else:
103
  raise ValueError(f"OCR engine '{ocr_engine}' not supported.")
104
 
models/yolov8n.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f59b3d833e2ff32e194b5bb8e08d211dc7c5bdf144b90d2c8412c47ccfc83b36
3
+ size 6549796