ahasera commited on
Commit
71e1889
·
1 Parent(s): 6a96152

fixed preprocessed result not shown and None return

Browse files
Files changed (1) hide show
  1. app.py +15 -2
app.py CHANGED
@@ -86,8 +86,21 @@ 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
- 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
 
 
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
+ return detections_raw, detections_preprocessed
104
  else:
105
  raise ValueError(f"OCR engine '{ocr_engine}' not supported.")
106