poocr解锁1s识别发票解析

poocr是基于腾讯云 OCR 接口的 Python 第三方库，通过二次封装将很多的复杂场景（身份证、发票、车牌等）的识别简化为「1 行代码」，支持图片 / URL 输入，兼容本地文件与在线图片，识别z准确率非常高。

1. 需求

最近有个需求,需要识别购车发票中的信息,然后将识别到的内容保存在excel表格中保存。找了网上很多的开源项目之后,发现poocr这个库非常的好用,于是决定在poocr这个库中添加功能,该功能可以识别图片中的指定类型的发票。

2. 前期准备

2.1 代码准备

查看poocr源码点击poocr查看gitcode的代码仓。
fork一个自己的代码仓

拉取代码

2.2 腾讯云接口

访问文字识别界面点击文字识别进入腾讯云文字识别界面。
查看腾讯云OCR识别功能

点击试用 -> 立即体验即可体验OCR识别功能。

开通功能并且领去免费试用机会搜索 文字识别,往下拉，点击开始使用。需要开通后付费设置免费使用机会才可以生效。

生成密钥ID和Key

点击新建密钥可以自动生成密钥，最多两个哦。

3. 功能实现

3.1 查看代码示例

从官方文档可知,调用腾讯云API时,首先需要有SecretId和SecretKey,对应上面申请的密钥;其次,在调用时需要使用models创建一个RecognizeGeneralInvoiceRequest的请求对象,使用这个封装请求的参数对象;最后,发送请求时,使用client调用RecognizeGeneralInvoice发送接口获取调用的结果。可以看出有两个参数是我们在使用过程中需要的:

ImageBase64: 图片的 Base64 值。
ImageUrl: 图片的 Url 地址。

注意:

支持的图片格式为PNG、JPG、JPEG、PDF，不支持 GIF 格式。 2. 支持的图片不超过 10M。 3. 图片下载时间不超过 3 秒。 4. 支持的图片像素：单边介于20-10000px之间。 5. 图片的 ImageUrl、ImageBase64 必须提供一个，如果都提供，只使用 ImageUrl。 6. 图中的`Types`参数如果指定,图片中的所有发票会按照你指定的类型来识别,而不是只识别图片中指定类型的发票。

3.2 查看poocr代码

从poocr代码结构中可知,api目录下的ocr2excel.py文件中,封装各种识别生成excel的方法。ocr.py文件中封装了各种识别生成json数据的方法。OCR.py文件中,通过models生成请求和client对腾讯云api主要调用方法的实现。

3.3 具体实现

在ocr2excel.py添加RecognizeGeneralInvoiceOCR2Excel方法,用于识别发票识别并且生成Excel文件,类别需要在接口文档中查看。

判断文件或者文件夹是否存在;
判断输出Excel路径是否合法;
批量识别图片,根据识别结果生成excel;

在ocr.py添加RecognizeGeneralInvoice方式,用于识别发票返回识别图片中的所有的json数据。

4. 功能测试

需要识别图片中火车票的数据并且生成Excel文件。代码执行结果: 识别结果: