PyTesseract 无法识别图像

Question

我目前面临 pytesseract 的问题，该软件无法检测到此图像中的数字：

https://i.stack.imgur.com/kmH2R.png

这是从应用了阈值过滤器的较大图像中获取的。

出于某种原因，pytesseract 不想识别此图像中的 6。有什么建议？这是我的代码：

image = #Insert raw image here. My code takes a screenshot.
image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
image = cv2.medianBlur(image, 3)
rel, gray = cv2.threshold(image, 127, 255, cv2.THRESH_BINARY)
# If you want to use the image from above, start here.
image = Image.fromarray(image)
string = pytesseract.image_to_string(image)
print(string)

编辑：通过进一步调查，我的代码在包含 2 位数字的情况下运行良好。但不是那些有单数的。

Answer 1

pytesseract 默认为查找大块文本的模式（PSM_SINGLE_BLOCK 或 --psm 6），为了让它检测到单个字符，您需要使用选项 --psm 10 (PSM_SINGLE_CHAR) 运行它。但是，由于您提供的图像角落中的黑点，它会将它们检测为随机破折号并且在此模式下不返回任何内容，因为它有多个字符，因此在这种情况下您需要使用 --psm 8 (PSM_SINGLE_WORD)：

string = pytesseract.image_to_string(image, config='--psm 8')

由此产生的输出将包括那些随机字符，因此您需要在 pytesseract 运行后剥离它们或改进数字周围的边界框以消除任何噪音。此外，如果检测到的所有字符都是数字，您可以在“--psm 8”之后添加“-c tessedit_char_whitelist=0123456789”以改进检测。

---

简化代码的其他一些小技巧是 cv2.imread 可以选择将图像读取为黑白，因此您无需在之后运行 cvtColor，只需执行以下操作：

image = cv2.imread('/path/to/image/6.png', 0)

您也可以在对 pytesseract 的调用中创建 PIL 图像对象，以便该行简化为：

string = pytesseract.image_to_string(Image.fromarray(img), config='--psm 8')

只要您的脚本顶部有“from PIL import Image”。