lamonov.pavel/PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

refs # Скрипт для вырезания VIN из СТС.
Some checks failed
Build/Publish Develop Docs / deploy (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Ламонов Павел
2025-07-22 19:27:36 +03:00
parent 644c61e9fe
commit 671c5085e9
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/detect_vin.py
View File

@@ -41,13 +41,16 @@ def auto_rotate_image(image):
ocr = PaddleOCR(lang='en', use_textline_orientation=True) ocr = PaddleOCR(lang='en', use_textline_orientation=True)
# Загрузка изображения # Загрузка изображения
image = cv2.imread("scripts/sts.jpg") image = cv2.imread("scripts/image.jpg")
# Автоматическое выравнивание # Автоматическое выравнивание
aligned_image = auto_rotate_image(image) # aligned_image = auto_rotate_image(image)
aligned_image = image
# Распознавание текста # Распознавание текста
result = ocr.predict(aligned_image) result = ocr.predict(image)
output_image = result[0]["doc_preprocessor_res"]["output_img"]
cv2.imwrite('scripts/output_image.jpg', output_image)
# Поиск VIN # Поиск VIN
vin_pattern = re.compile(r'^[A-HJ-NPR-Z0-9]{17}$') vin_pattern = re.compile(r'^[A-HJ-NPR-Z0-9]{17}$')
@@ -70,12 +73,12 @@ if found_vin:
x_max, y_max = bbox[2], bbox[3] x_max, y_max = bbox[2], bbox[3]
# Добавление запаса вокруг области # Добавление запаса вокруг области
vin_region = aligned_image[ vin_region = output_image[
y_min:y_max, y_min:y_max,
x_min:x_max x_min:x_max
] ]
cv2.imwrite('scripts/vin_rotated_corrected.jpg', vin_region) cv2.imwrite('scripts/vin.jpg', vin_region)
print(f"VIN найден: {found_vin}") print(f"VIN найден: {found_vin}")
else: else:
print("VIN не обнаружен после выравнивания") print("VIN не обнаружен после выравнивания")