Extrahieren Sie Text aus dem Bild mit MSER in OpenCV Python

Question

Ich möchte Text in einem Bild mit Mser erkennen und alle Nicht-Text-Regionen entfernen. Verwenden Sie den Code unten ich in der Lage war Text erkennen:

import cv2 
import sys 


mser = cv2.MSER_create() 
img = cv2.imread('signboard.jpg') 
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) 
vis = img.copy() 
regions, _ = mser.detectRegions(gray) 
hulls = [cv2.convexHull(p.reshape(-1, 1, 2)) for p in regions] 
cv2.polylines(vis, hulls, 1, (0, 255, 0)) 
cv2.imshow('img', vis) 
if cv2.waitKey(0) == 9: 
    cv2.destroyAllWindows() 

Wie kann ich alle Nicht-Textbereiche entfernen und um ein binäres Bild mit Text nur bekommen? Ich habe viel gesucht, konnte aber keinen Beispielcode finden, der dies mit Python und opencv macht.

Answer 1

Sie können ein binäres Bild mit den gefundenen Konturen erhalten. Zeichnen Sie die ausgefüllten Konturen einfach auf ein leeres Bild in weiß.

mask = np.zeros((img.shape[0], img.shape[1], 1), dtype=np.uint8) 
for contour in hulls: 
    cv2.drawContours(mask, [contour], -1, (255, 255, 255), -1) 

Hinweis: Siehe the official docs für mehr auf drawContours

Anschließend können Sie diese nur den Text zu extrahieren:

text_only = cv2.bitwise_and(img, img, mask=mask)