使用tesserocr库使用Python进行文本识别

时间:2018-08-10 05:17:28

标签: image opencv text ocr

我正在尝试从an image捕获文本,并且在下面的{script source)中使用此脚本

import cv2
import numpy as np
import tesserocr
from PIL import Image

# Path of working folder on Disk
src_path = "C:/Users/fyunu/OneDrive/Masaüstü/a.png"

def get_string(img_path):
    # Read image with opencv
    img = cv2.imread(img_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 = tesserocr.image_to_string(Image.open(src_path + "thres.png"))

    # Remove template file
    #os.remove(temp)

    return result


print ('--- Start recognize text from image ---')
print (get_string(src_path + "2.png"))

print ("------ Done -------")

但是,我遇到此错误: img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

“错误:OpenCV(3.4.1)C:\ Miniconda3 \ conda-bld \ opencv-suite_1533128839831 \ work \ modules \ imgproc \ src \ color.cpp:11147:错误:(-215)scn == 3 | | scn == 4在函数cv :: cvtColor“

如何解决此问题? 非常感谢。

0 个答案:

没有答案