从图像中提取特定的文本关联值

Question

我有一个图像，我想从图像中提取键和值对的详细信息。

例如，我想提取“MASTER-AIRWAYBILL NO:”的值

我已经写了使用 python opencv 和 OCR 从图像中提取整个文本，但我不知道如何从图像的整个结果文本中仅提取“MASTER-AIRWAYBILL NO:”的值。

请找到代码：

import cv2
import numpy as np
import pytesseract
from PIL import Image
print ("Hello")
src_path = "C:\\Users\Venkatraman.R\Desktop\\alpha_bill.jpg"
pytesseract.pytesseract.tesseract_cmd = r"C:\Program Files (x86)\Tesseract-OCR\tesseract.exe"


print (src_path)


# Read image with opencv
img = cv2.imread(src_path)

# Convert to gray
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# Apply dilation and erosion to remove some noise
kernel = np.ones((1, 1), np.uint8)
img = cv2.dilate(img, kernel, iterations=1)
img = cv2.erode(img, kernel, iterations=1)

# Write image after removed noise
cv2.imwrite(src_path + "removed_noise.png", img)

#  Apply threshold to get image with only black and white
#img = cv2.adaptiveThreshold(img, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 31, 2)

# Write the image after apply opencv to do some ...
cv2.imwrite(src_path + "thres.png", img)

# Recognize text with tesseract for python
result = pytesseract.image_to_string(Image.open(src_path + "thres.png"))


# Remove template file
#os.remove(temp)


print ('--- Start recognize text from image ---')
print (result)

所以输出应该是这样的：

主运单编号：157-46637194

Answer 1

您可以使用pytesseract image_to_string()和正则表达式来提取所需的文本，即：

\n\n

from PIL import Image\nimport pytesseract, re\nf = "ocr.jpg"\nt = pytesseract.image_to_string(Image.open(f))\nm = re.findall(r"MASTER-AIRWAYBILL NO: [\\d\xe2\x80\x94-]+", t)\nif m:\n    print(m[0])\n

\n\n\n\n

输出：

\n\n

MASTER-AIRWAYBILL NO: 157\xe2\x80\x9446637194\n

\n