很多軟件內置了OCR
功能,即圖片提取文字功能。有些是免費提供給大家使用,但有些是收費的。不管是免費的還是收費的,終究逃離不了隱私問題。用別人的OCR
,總得把圖片傳到對方的服務器。今天我們使用Python
開發一個OCR
軟件,如下圖所示。
本文基於PaddleOCR
搭建本地開發圖片提取文字軟件,因此需要安裝PaddlePaddle
環境。
如果您的機器有安裝CUDA9
或CUDA10
,推薦安裝GPU
版本的PaddlePaddle
,享受更快的運行速度。運行以下命令安裝:
python -m pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple
如果您的機器只有CPU
環境,運行速度會稍微慢一點。請運行以下命令安裝
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
有了PaddlePaddle
環境後,接下來安裝PaddleOCR庫,推薦使用2.0.1+版本:
pip install "paddleocr>=2.0.1"
注意:
對於
Windows
環境用戶:直接通過pip
安裝的shapely
庫可能出現[winRrror 126]
找不到指定模塊的問題。建議從這裡https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely下載shapely
安裝包完成安裝。
調用OCR
過程非常簡單,導入PaddleOCR
後,直接創建PaddleOCR
對象:
from paddleocr import PaddleOCR
ocr = PaddleOCR(use_angle_cls=True, lang="ch")
ocr.ocr(img_path, cls=True)
第2
行代碼中,use_angle_cls
參數用於確定是否使用角度分類模型,即是否識別垂直方向的文字。lang
參數表示識別的語言,我們傳入ch
,表示識別漢字。
第3
行代碼中, img_path
表示圖片路徑,cls
表示是否使用角度分類模型。
有了以上代碼就可以完成OCR
功能,但使用起來還不夠方便,我們進一步將OCR
功能封裝成軟件,便於交互。首先安裝PyQT5
:
pip install PyQt5
接下來調用PyQT5完成界面交互,篇幅原因,這裡只顯示調用ocr部分代碼,讀者可以直接拉到最後獲取完整源碼。
class OCRGUI(QWidget):
# 其他代碼略...
def run_ocr(self, img_path):
result = self.ocr.ocr(img_path, cls=True)
self.text.clear()
txts = [line[1][0] for line in result]
for txt in txts:
self.text.insertPlainText(txt + "\n")
self.loading.hide()
Python學習實戰
OCR
,獲取完整源碼。如果您覺得本文有幫助,辛苦您點個不需花錢的贊,您的舉手之勞將對我提供了無限的寫作動力! 也歡迎關注我的公眾號:Python學習實戰, 第一時間獲取最新文章。