-
Notifications
You must be signed in to change notification settings - Fork 0
/
tesseract.py
31 lines (27 loc) · 981 Bytes
/
tesseract.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
__author__ = 'David'
import os
import pytesseract
import Image
def retrieve_license_plate_from_img(bestandsnaam):
"""
Laad image bestand in, analyseer het en return het kenteken (als het gevonden is)
:param bestandsnaam: Bestandsnaam in de norm BESTANDSNAAM.jpg/png/bmp
:return: De tekst dat zich in het plaatje bevind, of object None of aan te geven dat het kenteken niet kan worden gevonden!
>>> retrieve_license_plate_from_img(TEST_IMAGE)
60-HT-RR
"""
try:
img = Image.open(bestandsnaam)
# Tesseract herkend het dash teken "-" niet correct, vandaar de .replace()
kenteken = pytesseract.image_to_string(img).replace("'", "-")
print("\tHet gevonden kenteken is", kenteken)
if kenteken == "":
return None
else:
return kenteken # TesseractOCR maakt van een - de tekst —
except FileNotFoundError as ex:
print(ex, ":", bestandsnaam)
print("Current path:", os.getcwd())
except Exception as ex:
print("Something terrible happened!")
print(ex)