这是一个简单的预处理步骤,用于在使用 pytesseract 之前清理图像
- 将图像转换为灰度
- 锐化图像
- 执行形态转换以增强文本
由于您的输入图像看起来很模糊,我们可以使用cv2.filter2D() 和通用锐化内核来锐化图像。其他类型的内核可以找到here
image = cv2.imread('1.png')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
sharpen_kernel = np.array([[-1,-1,-1], [-1,9,-1], [-1,-1,-1]])
sharpen = cv2.filter2D(gray, -1, sharpen_kernel)
文字有小孔,所以我们可以使用cv2.dilate()来封闭小孔并平滑图像
sharpen = 255 - sharpen
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (2,2))
dilate = cv2.dilate(sharpen, kernel, iterations=1)
result = 255 - dilate
结果如下。您可以尝试仅使用锐化图像或 pytesseract 增强图像
import cv2
import numpy as np
image = cv2.imread('1.png')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
sharpen_kernel = np.array([[-1,-1,-1], [-1,9,-1], [-1,-1,-1]])
sharpen = cv2.filter2D(gray, -1, sharpen_kernel)
cv2.imwrite('sharpen.png', sharpen)
sharpen = 255 - sharpen
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (2,2))
dilate = cv2.dilate(sharpen, kernel, iterations=1)
result = 255 - dilate
cv2.imwrite('result.png', result)
cv2.waitKey(0)