ahasera commited on
Commit
19fabdc
·
1 Parent(s): e3dc72a
Files changed (1) hide show
  1. app.py +3 -11
app.py CHANGED
@@ -86,19 +86,11 @@ def read_license_plate(license_plate_crop, ocr_engine='easyocr'):
86
  elif ocr_engine == 'paddleocr':
87
  result_raw = ocr_paddle.ocr(license_plate_crop)
88
  result_preprocessed = ocr_paddle.ocr(preprocessed_image)
89
-
90
- # no empty result
91
- if result_raw and result_raw[0]:
92
- detections_raw = [(None, result_raw[0][0][1][0], result_raw[0][0][1][1])]
93
- else:
94
- detections_raw = [(None, '', 0.0)]
95
-
96
- if result_preprocessed and result_preprocessed[0]:
97
- detections_preprocessed = [(None, result_preprocessed[0][0][1][0], result_preprocessed[0][0][1][1])]
98
- else:
99
- detections_preprocessed = [(None, '', 0.0)]
100
  else:
101
  raise ValueError(f"OCR engine '{ocr_engine}' not supported.")
 
102
 
103
  def extract_text(detections):
104
  plate = []
 
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
+
94
 
95
  def extract_text(detections):
96
  plate = []