まず、葉書の郵便番号部分を抽出する。
元画像
import cv2 import matplotlib.pyplot as plt def detect_zipno(fname): img = cv2.imread(fname) h, w = img.shape[:2] img = img[0:h//2, w//3:] gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) gray = cv2.GaussianBlur(gray, (3, 3), 0) im2 = cv2.threshold(gray, 140, 255, cv2.THRESH_BINARY_INV)[1] cnts = cv2.findContours(im2, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)[0] result = [] for pt in cnts: x, y, w, h = cv2.boundingRect(pt) if not(50 < w < 70): continue result.append([x, y, w, h]) result = sorted(result, key=lambda x: x[0]) result2 = [] lastx = -100 for x, y, w, h in result: if(x - lastx) < 10: continue result2.append([x, y, w, h]) lastx = x for x, y, w, h in result2: cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 3) return result2, img if __name__ == '__main__': cnts, img = detect_zipno("postcard.png") cv2.cvtColor(img, cv2.COLOR_BGR2RGB) cv2.imwrite("result.png", img)
$ python3 app.py
なんでやねん。なんで"2"と"4"が抽出されない。。。
前処理を少し変える。
img = img[0:h//5, w//3:] # h//3 -> h//5
4が抽出されない。
なぜだ?? ぼかしの周囲のサイズを変えてみる。
gray = cv2.GaussianBlur(gray, (1, 1), 0)
おおおおおおおおおおおおおおおおお
AI開発で、パラメータを調整するって、こういうこと????
うん、ちょっと興奮しました。
デバッグとはなんか感覚が違いますね。
あれ、というかこれ、記入する領域がわかってたらtesseractでOCRすりゃいいんだから、RPAできんじゃん。。
選挙システムとか注文書とか。。