【Python】PIL.Image 获取图像格式的二进制数据

one-ccs

已于 2022-03-31 23:33:38 修改

阅读量1.4k

点赞数

分类专栏： Python 文章标签： python 经验分享

于 2022-03-08 22:17:34 首次发布

本文为 one-ccs 原创文章，引用必须注明出处！

本文链接：https://blog.csdn.net/qq_43155814/article/details/123364653

版权

Python 专栏收录该内容

9 篇文章 1 订阅

订阅专栏

问题：屏幕截图进行 OCR 验证码识别需要图像格式的二进制数据，而 Image.tobyte() 获取的是无格式的二进制数据。

解决:

from io import BytesIO
import pyautogui
import ddddocr

screenshot = pyautogui.screenshot(region=[x + 0, y + 26, 100, 50])
# print(screenshot.tobytes())
# b'%\x1f\x1a%\x1f\x1a%\x1f\x1a%\x1f\x1a%\x1f\x1a%\x1f\x1a%\x1f\x1a%\x1f\x1a%\x1f...'
buffer = BytesIO()
screenshot.tobytes('raw', 'RGB')
# 以 PNG 格式保存在 buffer 中
screenshot.save(buffer, format='PNG')
# print(buffer.getvalue())
# b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00d\x00\x00\x002\x08\x02\x00\x00\x00%W\...'
code = ocr.classification(buffer.getvalue())