我正在尝试从以下图片中提取电话号码(调整大小后:)
我的代码:
from PIL import Image
from pyocr import pyocr
import pyocr.builders
import cStringIO
import os
os.putenv("TESSDATA_PREFIX", "/usr/share/tesseract-ocr/")
tools = pyocr.get_available_tools()
tool = tools[0]
langs = tool.get_available_languages()
lang = langs[0]
file = "test.png"
txt = tool.image_to_string(Image.open(file),
lang=lang,
builder=pyocr.builders.TextBuilder())
print txt
它返回空字符串。 当电话号码中没有( - )时,它会正确返回。 我该怎么办 ? 谢谢!
答案 0 :(得分:5)
好的,当我用tesseract和你提供的图像运行代码时,它完全返回了文本(包括破折号和空格)。那时你显然可以使用txt = txt.replace("-", "").replace(" ", "")
来摆脱破折号和空白。
Buuuuuut我知道OCR(即使我们都使用tesseract)在不同平台上会有所不同,所以我已经包含了我的评论建议的一个例子。
# I changed your imports a bit
from PIL import Image
from pyocr import pyocr
from pyocr import builders
import cStringIO
import os
# set up all your OCR stuff
os.putenv("TESSDATA_PREFIX", "/usr/share/tesseract-ocr/")
tools = pyocr.get_available_tools()
tool = tools[0]
langs = tool.get_available_languages()
lang = "eng" #set language to english to simplify things
# definte a function to return the text of a given image
def doOCR( fName ):
txt = tool.image_to_string(Image.open(fName), lang=lang, builder=builders.TextBuilder())
return txt
# define the path of the image we are going to read
path = "test.png"
# get the image dimensions
im = Image.open(path)
width, height = im.size
# define the points we want to split the image at
# these are the points where the dashes are
split_points = [119, 158]
# define the file names for the image parts
split_names = ["split-1.png", "split-2.png", "split-3.png"]
# define a function to crop the image and remove the dashes
def doCrop(imagePath, cropPath, x, y, x2, y2):
im = Image.open(imagePath)
box = (x, y, x2, y2)
region = im.crop(box) # extract the box region
region.save(cropPath) # save it as a separate image
# in the image you provided each "-" is ~10 pixels long
lenpix = 10
# crop the image at the split points
doCrop(path, split_names[0], 0, 0, split_points[0], height) # get the first section
doCrop(path, split_names[1], split_points[0] + lenpix, 0, split_points[1], height) # get the middle section
doCrop(path, split_names[2], split_points[1] + lenpix, 0, width, height) # get the final section
# define a variable for our final value
finalValue = ""
# finally iterate through split files
# and add the OCR results from each split together
for f in split_names:
finalValue += doOCR(f) # concatenate the ocr value with the final
os.remove(f) # remove the split file now that we've used it
# display the final value
print finalValue
对我来说就像一个魅力:
希望这有帮助!