I have a code that detects and reads license plates from images and videos. But the plates it detects are from the UK, and I would like to change the format of the plates detected to fit the Moroccan license plates (that are written like this : XXXXX| ب | XX )
The code I have tries to see if the detected text fits the license plate format so that it can approve it:
CODE
# Initialize the OCR readerreader = easyocr.Reader(['en'], gpu=False)# Mapping dictionaries for character conversiondict_char_to_int = {'O': '0','I': '1','J': '3','A': '4','G': '6','S': '5'}dict_int_to_char = {'0': 'O','1': 'I','3': 'J','4': 'A','6': 'G','5': 'S'}def license_complies_format(text):""" Check if the license plate text complies with the required format. Args: text (str): License plate text. Returns: bool: True if the license plate complies with the format, False otherwise. # if it in the char dict we aslo return it and will replace it later""" if len(text) != 7: return False if (text[0] in string.ascii_uppercase or text[0] in dict_int_to_char.keys()) and \ (text[1] in string.ascii_uppercase or text[1] in dict_int_to_char.keys()) and \ (text[2] in ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'] or text[2] in dict_char_to_int.keys()) and \ (text[3] in ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'] or text[3] in dict_char_to_int.keys()) and \ (text[4] in string.ascii_uppercase or text[4] in dict_int_to_char.keys()) and \ (text[5] in string.ascii_uppercase or text[5] in dict_int_to_char.keys()) and \ (text[6] in string.ascii_uppercase or text[6] in dict_int_to_char.keys()): return True else: return Falsedef format_license(text):""" Format the license plate text by converting characters using the mapping dictionaries. Args: text (str): License plate text. Returns: str: Formatted license plate text.""" license_plate_ = '' mapping = {0: dict_int_to_char, 1: dict_int_to_char, 4: dict_int_to_char, 5: dict_int_to_char, 6: dict_int_to_char, 2: dict_char_to_int, 3: dict_char_to_int} for j in [0, 1, 2, 3, 4, 5, 6]: if text[j] in mapping[j].keys(): license_plate_ += mapping[j][text[j]] else: license_plate_ += text[j] return license_plate_# the 2 most importantdef read_license_plate(license_plate_crop):""" Read the license plate text from the given cropped image. Args: license_plate_crop (PIL.Image.Image): Cropped image containing the license plate. Returns: tuple: Tuple containing the formatted license plate text and its confidence score.""" detections = reader.readtext(license_plate_crop) for detection in detections: bbox, text, score = detection text = text.upper().replace('', '') if license_complies_format(text): return format_license(text), score return None, NoneI tried changing the format in the function license_complies_format, but I never managed to make it work.