python圖片驗證碼識別最新模塊muggle_ocr

一.官方文檔

https://pypi.org/project/muggle-ocr/

二模塊安裝

pip install muggle-ocr
# 因模塊過新，阿里/清華等第三方源可能尚未更新鏡像，因此手動指定使用境外源，為了提高依賴的安裝速度，可預先自行安裝依賴：tensorflow/numpy/opencv-python/pillow/pyyaml

三.使用代碼

# 導入包
import muggle_ocr

# 初始化；model_type 包含了 ModelType.OCR/ModelType.Captcha 兩種,OCR能識別中文
sdk = muggle_ocr.SDK(model_type=muggle_ocr.ModelType.OCR)

with open(r"test1.png", "rb") as f:
    b = f.read()
text = sdk.predict(image_bytes=b)
print(text)


# ModelType.Captcha 可識別4-6位驗證碼
sdk = muggle_ocr.SDK(model_type=muggle_ocr.ModelType.Captcha)
with open(r"test1.png", "rb") as f:
    b = f.read()
text = sdk.predict(image_bytes=b)
print(text)