是否可以对字节图像进行 pytesseract?

Is it possible to pytesseract a bytes image?

我正在尝试使用 cv2 裁剪图像(将其转换为字节文件,因此不需要保存)然后执行 pytesseract。

这样我就不需要在过程中保存图像两次了。

  1. 首先创建图像
  2. 裁剪图像时

处理中...

## CROPPING THE IMAGE REGION

ys, xs = np.nonzero(mask2)
ymin, ymax = ys.min(), ys.max()
xmin, xmax = xs.min(), xs.max()

croped = image[ymin:ymax, xmin:xmax]


pts = np.int32([[xmin, ymin],[xmin,ymax],[xmax,ymax],[xmax,ymin]])
cv2.drawContours(image, [pts], -1, (0,255,0), 1, cv2.LINE_AA)
#OPENCV IMAGE TO BYTES WITHOUT SAVING TO DISK

is_success, im_buf_arr = cv2.imencode(".jpg", croped)
byte_im = im_buf_arr.tobytes()
#PYTESSERACT IMAGE USING A BYTES FILE

Results = pytesseract.image_to_string(byte_im, lang="eng")
print(Results)

不幸的是我收到错误消息:不支持的图像对象

我错过了什么吗?有没有办法在裁剪时不需要保存文件来完成这个过程?非常感谢任何帮助。

from PIL import Image

img_tesseract = Image.fromarray(croped)
Results = pytesseract.image_to_string(img_tesseract, lang="eng")

你有 croped 这是一个 numpy 数组。

根据 pytesseract examples,您只需这样做:

# tesseract needs the right channel order
cropped_rgb = cv2.cvtColor(croped, cv2.COLOR_BGR2RGB)

# give the numpy array directly to pytesseract, no PIL or other acrobatics necessary
Results = pytesseract.image_to_string(cropped_rgb, lang="eng")

from PIL import Image
import io
def bytes_to_image(image_bytes):
    io_bytes = io.BytesIO(image_bytes)
    return Image.open(io_bytes)
pytesseract.image_to_data(byte_array_image,lang='eng')