如果覺得文章寫得好,想要博客文章中的數據,請關注公眾號:【Z先生點記】,已經為你准備了 50本+ Python 電子書籍 與 200G + 優質視頻資料,后台回復關鍵字:1024 即可獲取;添加作者【個人微信】,可與作者直接進行交流,
文字 OCR 識別技術現在已經相當成熟了,無論 其 准確度還是識別速度 都能夠滿足我們的日常需要;今天給大家介紹一個 Python 包,該包的主要功能就是用於 OCR 識別的,包的名字叫 Pyteeseract,借助這個包幾行代碼就能快速識別一張文本圖片
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/
提供有 3.0+,4.0+及5.0+版本使用,下載完之后安裝(安裝方式就是傻瓜式安裝)
tesseract 安裝成功之后,需把存放 tesseract.exe 的文件路徑加入到環境變量中,如下圖我的 tesseract.exe 存放的文件夾就是 F:/Program Files/Tesseract-OCR 加入環境變量即可;
2,pip install pytesseract
在命令行中,用 pip 工具下載 pytesseract 包
pip install pyteeseract
3,修改pytesseract.py 腳本
在 步驟 2 的基礎之上,找到 pytesseract 的安裝路徑,如果 Python 是通過 Anaconda 安裝的話的話,安裝路徑一般都在 Anaconda/Lib/site-packages 文件夾下;找到之后找 pytesseract 文件夾下的 pytesseract.py 腳本文件,
找到之后,用記事本打開 pyresseract.py,通過 ctrl +f 快速搜索功能定位 tesseract_cmd ,修改后面的文件路徑信息(用上面提到的 tesseract.exe 安裝路徑進行替換即可);
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)
效果預覽,識別前
識別后
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 即可
效果預覽,識別前
識別后,對於中文來說識別效果並不是太好,猜測可是是版本的原因:
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
2,估計識別出來每個字符的邊框信息,圖片中的位置分辨率范圍:
print(pytesseract.image_to_boxes(img_path,lang = 'chi_sim',config= tessdata_dir_config))
3,關於 pyteeseract 還有很多用法還未介紹,有興趣的小伙伴可去官網介紹進行了解,鏈接如下: