基于Tesseract与Opencv的电子发票识别[1]

本文我们将尝试使用tesseract识别电子发票上的信息并不断提高识别准确率,是一个逐渐调整的过程,仅用于记录研究过程。

  1. 图像识别:使用tesseract识别。
  2. 图像预处理:使用OpenCV等图像处理库对发票图像进行预处理,如灰度化、去噪、二值化等,以提高文字清晰度。
  3. 语言和数据集定制:根据具体需求训练Tesseract使用特定语言和数据集,特别是对于包含大量特定格式和术语的电子发票。

目录

1. 基于tesseract识别电子发票

2. 借助OpenCV对图像进行预处理

2.1 读取图像

2.2 转换为灰度图像

2.3 高斯模糊降噪

2.4 二值化处理

2.5 Opencv预处理代码与调优[批处理]

2.5.1 完整代码

2.5.2 调整后批处理代码


1. 基于tesseract识别电子发票

tesseract安装参考:Tesseract OCR 的使用

 确认安装成功:

 tesseract 命令行识别命令参考:tesseract  图片路径  识别结果  -l  使用的字符集

tesseract 0008.jpg result -l chi_sim
  •  0008.jpg  是需要识别的图片
  •  result 是识别结果,识别完成后生成result.txt
  • chi_sim 是我们使用的识别库,基于chi_sim.traineddata识别,横排汉字

 识别参考:

识别结果部分如下:

2. 借助OpenCV对图像进行预处理

OpenCV(Open Source Computer Vision Library)是一个开源的计算机视觉和机器学习软件库,它包含了大量的图像处理和计算机视觉算法。OpenCV最初由英特尔公司于1999年发起,现在由多个组织维护,包括Willow Garage和Intel。它旨在推动计算机视觉领域的实时应用,是一个跨平台的开源库,支持多种编程语言,如C、C++、Python、Ruby和Java。

2.1 读取图像

使用cv2.imread()函数读取发票图片。

import cv2

# 读取图像
img = cv2.imread("0001.jpg")

判断图像获取是否成功,成功获取则打开图片查看

#检查图片是否成功读取
if img is None:
    print("图片读取失败,请检查路径是否正确!")
else:
    # 显示图片
    cv2.imshow('图片', img)
    # 等待用户按键,0表示无限等待,直到用户按下任意键
    cv2.waitKey(0)
    # 关闭所有OpenCV窗口
    cv2.destroyAllWindows()

图片太大的问题,可以提前增加命令,调整窗口大小

# 调整窗口大小
cv2.namedWindow('图片', cv2.WINDOW_NORMAL)
cv2.resizeWindow('图片', 800, 600)

2.2 转换为灰度图像

将彩色图像转换为灰度图像,不仅简化了图像信息,还增强了文字的对比度,使得后续文字识别更加高效和准确。

# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

cvtColor()方法是OpenCV库中用于图像颜色空间转换的函数,它支持多种颜色空间的相互转换,如BGR到灰度图、BGR到HSV等。cv2.cvtColor(img, cv2.COLOR_BGR2GRAY),使用cvtColor()方法将图片img从BGR颜色空间转换为了灰度图。

2.3 高斯模糊降噪

使用高斯模糊来减少图像中的噪声。使用高斯降噪的原因:消除噪声对OCR的干扰,平衡去噪与边缘保留。

例如,发票上的细小划痕或墨点经高斯滤波后,会被均匀分散为低强度信号,减少对文字特征的干扰。

# 高斯模糊降噪
blurred = cv2.GaussianBlur(gray, (5, 5), 0)

2.4 二值化处理

将图像转换为二值图像,以便于字符分割和识别。

二值化处理的核心优势在于:(1)增强文字与背景对比度,通过将彩色或灰度图像转换为黑白二值图像,可显著提升文字区域与背景的区分度。(2)降低计算复杂度,二值化将像素值简化为0(黑)和255(白),相比原始RGB图像的每个像素24位数据量,存储和处理效率提升约24倍。(3)提升特征提取精度,黑白分明的图像结构更有利于关键区域的定位与分割。

# 二值化处理
binary = cv2.threshold(blurred, 127, 255, cv2.THRESH_BINARY)

2.5 Opencv预处理代码与调优[批处理]

2.5.1 完整代码

import cv2

# 读取图像
img = cv2.imread("0080.jpg")

# 调整窗口大小
cv2.namedWindow('图片', cv2.WINDOW_NORMAL | cv2.WINDOW_KEEPRATIO)
cv2.resizeWindow('图片', 800, 600)

# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 高斯模糊降噪
blurred = cv2.GaussianBlur(gray, (5, 5), 0)

# 二值化处理
_,binary = cv2.threshold(blurred, 180, 255, cv2.THRESH_BINARY)
#_,binary = cv2.threshold(blurred, 180, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C)

# 输出图片
cv2.imshow('图片', rotated)
cv2.waitKey(0)
cv2.destroyAllWindows()
cv2.imwrite('output.bmp', rotated)

2.5.2 调整后批处理代码

import cv2
import os

def process_images(input_folder, output_folder):
    # 创建输出文件夹(若不存在)
    os.makedirs(output_folder, exist_ok=True)
    
    # 获取所有图片文件
    valid_ext = ('.jpg', '.jpeg', '.png', '.bmp')
    image_files = [f for f in os.listdir(input_folder) 
                  if f.lower().endswith(valid_ext)]
    
    for filename in image_files:
        # 读取原始图片
        img_path = os.path.join(input_folder, filename)
        image = cv2.imread(img_path)
        
        if image is not None:
            try:
                # 图像处理示例(可根据需求修改)
                gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
                blurred = cv2.GaussianBlur(gray, (5, 5), 0)
                _,binary = cv2.threshold(blurred, 180, 255, cv2.THRESH_BINARY)
                
                # 构建输出路径
                base_name = os.path.splitext(filename)[0]
                output_path = os.path.join(output_folder, f"{base_name}.jpg")
                
                # 保存处理结果
                if cv2.imwrite(output_path, binary):
                    print(f"成功保存: {output_path}")
                else:
                    print(f"保存失败: {filename}")
                    
            except Exception as e:
                print(f"处理异常 {filename}: {str(e)}")
        else:
            print(f"读取失败: {filename}")

if __name__ == "__main__":
    input_dir = "input"  # 输入图片文件夹
    output_dir = "output"  # 输出文件夹
    process_images(input_dir, output_dir)

测试成果:部分图片预处理之前识别不到发票号,预处理之后可以识别发票号。

发现问题:部分图片识别到的发票号码会有乱码的情况。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值