46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
"""
|
|
exec(open("scripts/detect_vin.py").read())
|
|
"""
|
|
|
|
from paddleocr import PaddleOCR
|
|
import cv2
|
|
import re
|
|
|
|
ocr = PaddleOCR(lang='en', use_textline_orientation=True)
|
|
|
|
def cut_vin(ocr, input_image, output_image):
|
|
"""
|
|
Вырезать VIN из изображения.
|
|
"""
|
|
|
|
image = cv2.imread(input_image)
|
|
|
|
result = ocr.predict(image)
|
|
processed_image = result[0]["doc_preprocessor_res"]["output_img"]
|
|
|
|
vin_pattern = re.compile(r'^[A-HJ-NPR-Z0-9]{17}$')
|
|
found_vin = None
|
|
|
|
for text, bbox in zip(result[0]["rec_texts"], result[0]["rec_boxes"]):
|
|
if vin_pattern.match(text):
|
|
found_vin = text
|
|
break
|
|
|
|
if found_vin:
|
|
x_min, y_min = bbox[0], bbox[1]
|
|
x_max, y_max = bbox[2], bbox[3]
|
|
|
|
vin_region = processed_image[y_min:y_max, x_min:x_max]
|
|
|
|
cv2.imwrite(output_image, vin_region)
|
|
|
|
return found_vin
|
|
|
|
vin = cut_vin(ocr, "input/image.jpg", "output/vin.jpg")
|
|
|
|
if vin:
|
|
print(f"VIN найден: {vin}.")
|
|
else:
|
|
print("VIN не обнаружен.")
|
|
|