python opencv將表格圖片按照表格框線分割和識別
阿新 • • 發佈:2020-01-09
如下小程式為使用python+opencv將表格圖片,按照表格進行分割,並識別分割後的子圖片中的文字,希望對需要的小夥伴有一些些幫助。具體的實現見如下程式碼。
# -*- coding: utf-8 -*- """ Created on Tue May 28 19:23:19 2019 將圖片按照表格框線交叉點分割成子圖片(傳入圖片路徑) @author: hx """ import cv2 import numpy as np import pytesseract image = cv2.imread('C:/Users/Administrator/Desktop/7.jpg',1) #灰度圖片 gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY) #二值化 binary = cv2.adaptiveThreshold(~gray,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,cv2.THRESH_BINARY,35,-5) #ret,binary = cv2.threshold(~gray,127,cv2.THRESH_BINARY) cv2.imshow("二值化圖片:",binary) #展示圖片 cv2.waitKey(0) rows,cols=binary.shape scale = 40 #識別橫線 kernel = cv2.getStructuringElement(cv2.MORPH_RECT,(cols//scale,1)) eroded = cv2.erode(binary,kernel,iterations = 1) #cv2.imshow("Eroded Image",eroded) dilatedcol = cv2.dilate(eroded,iterations = 1) cv2.imshow("表格橫線展示:",dilatedcol) cv2.waitKey(0) #識別豎線 scale = 20 kernel = cv2.getStructuringElement(cv2.MORPH_RECT,(1,rows//scale)) eroded = cv2.erode(binary,iterations = 1) dilatedrow = cv2.dilate(eroded,iterations = 1) cv2.imshow("表格豎線展示:",dilatedrow) cv2.waitKey(0) #標識交點 bitwiseAnd = cv2.bitwise_and(dilatedcol,dilatedrow) cv2.imshow("表格交點展示:",bitwiseAnd) cv2.waitKey(0) # cv2.imwrite("my.png",bitwiseAnd) #將二值畫素點生成圖片儲存 #標識表格 merge = cv2.add(dilatedcol,dilatedrow) cv2.imshow("表格整體展示:",merge) cv2.waitKey(0) #兩張圖片進行減法運算,去掉表格框線 merge2 = cv2.subtract(binary,merge) cv2.imshow("圖片去掉表格框線展示:",merge2) cv2.waitKey(0) #識別黑白圖中的白色交叉點,將橫縱座標取出 ys,xs = np.where(bitwiseAnd>0) mylisty=[] #縱座標 mylistx=[] #橫座標 #通過排序,獲取跳變的x和y的值,說明是交點,否則交點會有好多畫素值值相近,我只取相近值的最後一點 #這個10的跳變不是固定的,根據不同的圖片會有微調,基本上為單元格表格的高度(y座標跳變)和長度(x座標跳變) i = 0 myxs=np.sort(xs) for i in range(len(myxs)-1): if(myxs[i+1]-myxs[i]>10): mylistx.append(myxs[i]) i=i+1 mylistx.append(myxs[i]) #要將最後一個點加入 i = 0 myys=np.sort(ys) #print(np.sort(ys)) for i in range(len(myys)-1): if(myys[i+1]-myys[i]>10): mylisty.append(myys[i]) i=i+1 mylisty.append(myys[i]) #要將最後一個點加入 print('mylisty',mylisty) print('mylistx',mylistx) #迴圈y座標,x座標分割表格 for i in range(len(mylisty)-1): for j in range(len(mylistx)-1): #在分割時,第一個引數為y座標,第二個引數為x座標 ROI = image[mylisty[i]+3:mylisty[i+1]-3,mylistx[j]:mylistx[j+1]-3] #減去3的原因是由於我縮小ROI範圍 cv2.imshow("分割後子圖片展示:",ROI) cv2.waitKey(0) #special_char_list = '`~!@#$%^&*()-_=+[]{}|\\;:‘',。《》/?ˇ' pytesseract.pytesseract.tesseract_cmd = 'E:/Tesseract-OCR/tesseract.exe' text1 = pytesseract.image_to_string(ROI) #讀取文字,此為預設英文 #text2 = ''.join([char for char in text2 if char not in special_char_list]) print('識別分割子圖片資訊為:'+text1) j=j+1 i=i+1
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。