一個 Python 包 pytesseract ,幾行代碼實現 OCR 文本識別技術!


如果覺得文章寫得好,想要博客文章中的數據,請關注公眾號:【Z先生點記】,已經為你准備了 50本+ Python 電子書籍 與 200G + 優質視頻資料,后台回復關鍵字:1024 即可獲取;添加作者【個人微信】,可與作者直接進行交流,

文字 OCR 識別技術現在已經相當成熟了,無論 其 准確度還是識別速度 都能夠滿足我們的日常需要;今天給大家介紹一個 Python 包,該包的主要功能就是用於 OCR 識別的,包的名字叫 Pyteeseract,借助這個包幾行代碼就能快速識別一張文本圖片

pytesseract.jpg

Pytesseract 包是由 開源工具 Tesseract 得到的,由 Hewlett Packard 實驗室開發,在 2005 年實行開源;自2006 年之后由谷歌和一些優秀的開源貢獻者共同開發維護

Tesseract 在 3.x 版本之后逐漸成熟,支持多種圖片格式並且逐步加入多語言文本識別;但 Tesseract 3.x 版本依舊 基於傳統計算機視覺算法,在過去的幾年得益於 Deep Learning 的快速迭代,無論是准確率與速度方面都要優於傳統算法;在 4.0 版本之后 Tesseract 加入了 Deep Learning 模塊, 是基於 Recognition 的 LSTM,而 LSTM 就是可歸類為 RNN(循環卷積神經網絡);

本篇文章的實驗是基於 Tesseract3.05 版本實現的,最后在中文語言識別方面准確率稍次,可能是因為沒有使用4.0+的原因在,后來才了解已經有 4.0+ 版本甚至 5.0+ (但不太穩定)且都是基於 Deep Learning 模塊的,但因為太懶就不想改了,,,

先交代一下實驗環境:

  • os: Win10;

  • Python 3.8;

  • pyteeseract 0.3.8;

  • Tesseract 3.05;

pyteeseract 安裝

1,安裝 tesseract 工具

相對其它程序包,pyteeseract 的安裝步驟會相對繁瑣一點,因為 pyteeseract 識別功能是基於 tesseract 開源工具完成的,所以第一步 安裝 tesseract ,安裝包下載鏈接:

https://digi.bib.uni-mannheim.de/tesseract/

Snipaste_2020-09-12_14-30-55.jpg

提供有 3.0+,4.0+及5.0+版本使用,下載完之后安裝(安裝方式就是傻瓜式安裝)

tesseract 安裝成功之后,需把存放 tesseract.exe 的文件路徑加入到環境變量中,如下圖我的 tesseract.exe 存放的文件夾就是 F:/Program Files/Tesseract-OCR 加入環境變量即可;

Snipaste_2020-09-12_14-35-04.jpg

2,pip install pytesseract

在命令行中,用 pip 工具下載 pytesseract 包

pip install pyteeseract

3,修改pytesseract.py 腳本

在 步驟 2 的基礎之上,找到 pytesseract 的安裝路徑,如果 Python 是通過 Anaconda 安裝的話的話,安裝路徑一般都在 Anaconda/Lib/site-packages 文件夾下;找到之后找 pytesseract 文件夾下的 pytesseract.py 腳本文件,

Snipaste_2020-09-12_15-54-20.jpg

找到之后,用記事本打開 pyresseract.py,通過 ctrl +f 快速搜索功能定位 tesseract_cmd ,修改后面的文件路徑信息(用上面提到的 tesseract.exe 安裝路徑進行替換即可);

Snipaste_2020-09-12_15-54-30.jpg

2, pytesseract 使用

程序包的用法也相對比較簡單,幾行代碼就能搞定,下面代碼就是把一張圖片中的文字識別,並轉化為 字符串打印出來,選擇識別語言 英語( 更改 lang = ‘eng' 參數即可)

import pytesseract
import cv2

img_path = "G:/Coding/One_hundred_days/Data/orc_image2.jpg"

# 下面一行代碼很重要
tessdata_dir_config = '--tessdata-dir "F://Program Files//Tesseract-OCR//tessdata"'

im = cv2.imread(img_path)
img = cv2.cvtColor(im,cv2.COLOR_BGR2RGB)

text = pytesseract.image_to_string(img,lang= 'eng',config= tessdata_dir_config,)
print(text)

效果預覽,識別前

orc_image1.jpg

識別后

Snipaste_2020-09-12_12-54-01.png

pytesseract 支持將 OpenCV 和 PIL 讀取后的圖像作為輸入,但圖像格式需為 RGB 模式,因此 OpenCV 讀取之后還要加入一行代碼把圖像的 BGR 模式轉化為 RGB

另外需要注意一點 ,上面實例中下面這一行代碼不能去掉(用於后面 image_to_string() 函數中的 config 參數的設置)

tessdata_dir_config = '--tessdata-dir "F://Program Files//Tesseract-OCR//tessdata"'

否則會報下面的錯誤,tessdata 文件路徑定位失敗

Failed loading language 'eng' Tesseract couldn't load any languages! Could not initialize tesseract.')

tessdata 文件路徑存放的是語言包文件,是用於 識別圖像中不同語言,通過修改 lang 參數來進行設定;但需要知道的是,tesseract 工具起初默認語言為 eng(英文),若需要借助 tesseract 識別不同語種需要下載對應的 語言包文件,安裝到 tessdata 文件夾下即可

例如上面案例中我用的是 英語,這里我想識別圖片中的中文字符,就需要把中文語言包下載 到 testdata 中,各語言包的下載地址,https://github.com/tesseract-ocr/tessdata

再把代碼中 image_to_string() 中 lang 參數設為 chi_sim 即可

效果預覽,識別前

orc_image2.jpg

識別后,對於中文來說識別效果並不是太好,猜測可是是版本的原因:

Snipaste_2020-09-12_15-57-06.jpg

pyteeseract 其他用法

1,除了上面可以直接把圖像中內容識別轉化為字符串之外,還可直接轉化 pdf 文件形式導出

# Get a searchable PDF
pdf = pytesseract.image_to_pdf_or_hocr('test.png', extension='pdf')
with open('test.pdf', 'w+b') as f:
    f.write(pdf) # pdf type is bytes by default

Snipaste_2020-09-12_15-35-49.jpg

2,估計識別出來每個字符的邊框信息,圖片中的位置分辨率范圍:

print(pytesseract.image_to_boxes(img_path,lang = 'chi_sim',config= tessdata_dir_config))

Snipaste_2020-09-12_15-38-26.jpg

3,關於 pyteeseract 還有很多用法還未介紹,有興趣的小伙伴可去官網介紹進行了解,鏈接如下:

https://pypi.org/project/pytesseract/


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM