python opencv+pytesseract 驗證碼識別

文章目錄

• 一、環境配置
• 二、驗證碼識別
• 實體1
• 實體2
• 實體3

一、環境配置

• 需要 pillow 和 pytesseract 這兩個庫，pip install 安裝就好了，

pip install pillow -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install pytesseract -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

• 1
• 2

• 安裝好Tesseract-OCR.exe
• pytesseract 庫的配置：搜索找到pytesseract.py，打開該.py檔案，找到 tesseract_cmd，改變它的值為剛才安裝 tesseract.exe 的路徑，

二、驗證碼識別

識別驗證碼，需要先對影像進行預處理，去除會影響識別準確度的線潭訓噪點，提高識別準確度，

實體1

import cv2 as cv
import pytesseract
from PIL import Image


def recognize_text(image):
    # 邊緣保留濾波  去噪
    dst = cv.pyrMeanShiftFiltering(image, sp=10, sr=150)
    # 灰度影像
    gray = cv.cvtColor(dst, cv.COLOR_BGR2GRAY)
    # 二值化
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)
    # 形態學操作   腐蝕  膨脹
    erode = cv.erode(binary, None, iterations=2)
    dilate = cv.dilate(erode, None, iterations=1)
    cv.imshow('dilate', dilate)
    # 邏輯運算  讓背景為白色  字體為黑  便于識別
    cv.bitwise_not(dilate, dilate)
    cv.imshow('binary-image', dilate)
    # 識別
    test_message = Image.fromarray(dilate)
    text = pytesseract.image_to_string(test_message)
    print(f'識別結果：{text}')


src = cv.imread(r'./test/044.png')
cv.imshow('input image', src)
recognize_text(src)
cv.waitKey(0)
cv.destroyAllWindows()

運行效果如下：

識別結果：3n3D

Process finished with exit code 0

實體2

import cv2 as cv
import pytesseract
from PIL import Image


def recognize_text(image):
    # 邊緣保留濾波  去噪
    blur =cv.pyrMeanShiftFiltering(image, sp=8, sr=60)
    cv.imshow('dst', blur)
    # 灰度影像
    gray = cv.cvtColor(blur, cv.COLOR_BGR2GRAY)
    # 二值化
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)
    print(f'二值化自適應閾值：{ret}')
    cv.imshow('binary', binary)
    # 形態學操作  獲取結構元素  開操作
    kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 2))
    bin1 = cv.morphologyEx(binary, cv.MORPH_OPEN, kernel)
    cv.imshow('bin1', bin1)
    kernel = cv.getStructuringElement(cv.MORPH_OPEN, (2, 3))
    bin2 = cv.morphologyEx(bin1, cv.MORPH_OPEN, kernel)
    cv.imshow('bin2', bin2)
    # 邏輯運算  讓背景為白色  字體為黑  便于識別
    cv.bitwise_not(bin2, bin2)
    cv.imshow('binary-image', bin2)
    # 識別
    test_message = Image.fromarray(bin2)
    text = pytesseract.image_to_string(test_message)
    print(f'識別結果：{text}')


src = cv.imread(r'./test/045.png')
cv.imshow('input image', src)
recognize_text(src)
cv.waitKey(0)
cv.destroyAllWindows()

運行效果如下：

二值化自適應閾值：181.0
識別結果：8A62N1

Process finished with exit code 0

• 1
• 2
• 3
• 4

實體3

import cv2 as cv
import pytesseract
from PIL import Image


def recognize_text(image):
    # 邊緣保留濾波  去噪
    blur = cv.pyrMeanShiftFiltering(image, sp=8, sr=60)
    cv.imshow('dst', blur)
    # 灰度影像
    gray = cv.cvtColor(blur, cv.COLOR_BGR2GRAY)
    # 二值化  設定閾值  自適應閾值的話 黃色的4會提取不出來
    ret, binary = cv.threshold(gray, 185, 255, cv.THRESH_BINARY_INV)
    print(f'二值化設定的閾值：{ret}')
    cv.imshow('binary', binary)
    # 邏輯運算  讓背景為白色  字體為黑  便于識別
    cv.bitwise_not(binary, binary)
    cv.imshow('bg_image', binary)
    # 識別
    test_message = Image.fromarray(binary)
    text = pytesseract.image_to_string(test_message)
    print(f'識別結果：{text}')


src = cv.imread(r'./test/045.jpg')
cv.imshow('input image', src)
recognize_text(src)
cv.waitKey(0)
cv.destroyAllWindows()

運行效果如下：

二值化設定的閾值：185.0
識別結果：7364

Process finished with exit code 0

本文首發于python 黑洞網，博客園同步更新

標籤：Python

上一篇：我發現這個up封面確實有點東西，爬取B站視頻的封面圖片

下一篇：Python資料可視化：象限圖的應用