OpenCV:如何从背景中删除文本

Sha*_*ank 4 python opencv image image-processing computer-vision

现在我正在尝试创建一个程序,它从背景中删除文本,但我在通过它时遇到了很多问题

我的方法是使用 pytesseract 来获取文本框,一旦我得到文本框,我就使用 cv2.inpaint 来绘制它并从那里删除文本。简而言之:

d = pytesseract.image_to_data(img, output_type=Output.DICT) # Get text
n_boxes = len(d['level']) # get boxes
for i in range(n_boxes): #  Looping through boxes
    # Get coordinates
    (x, y, w, h) = (d['left'][i], d['top'][i], d['width'][i], d['height'][i])
    crop_img = img[y:y+h, x:x+w] # Crop image
    gray = cv2.cvtColor(crop_img, cv2.COLOR_BGR2GRAY)
    gray = inverte(gray) # Inverse it
    thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU)[1]
    dst = cv2.inpaint(crop_img, thresh, 10, cv2.INPAINT_TELEA) # Then Inpaint
    img[y:y+h, x:x+w] = dst # Place back cropped image back to the source image
Run Code Online (Sandbox Code Playgroud)

现在的问题是我无法完全删除文本图像: 在此处输入图片说明

现在我不确定我可以使用什么其他方法从图像中删除文本,我对此很陌生,这就是我面临问题的原因。任何帮助深表感谢

注意:图像看起来被拉伸了,因为我调整了它的大小以显示在屏幕大小

原图:

在此处输入图片说明

nat*_*ncy 6

这是使用形态学操作+轮廓过滤的方法

  • 将图像转换为灰度
  • 获得二值图像的大津阈值
  • 执行 morph close 以将单词连接成单个轮廓
  • 展开以确保所有文本位都包含在轮廓中
  • 使用轮廓区域查找轮廓并过滤
  • 通过使用背景颜色在轮廓矩形中“填充”来删除文本

我使用 chrome 开发人员工具来确定图像的背景颜色(222,228,251)。如果要动态确定背景颜色,可以尝试使用 k-means 查找主色。这是结果

在此处输入图片说明

import cv2

image = cv2.imread('1.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]

close_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (15,3))
close = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, close_kernel, iterations=1)

dilate_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5,3))
dilate = cv2.dilate(close, dilate_kernel, iterations=1)

cnts = cv2.findContours(dilate, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]
for c in cnts:
    area = cv2.contourArea(c)
    if area > 800 and area < 15000:
        x,y,w,h = cv2.boundingRect(c)
        cv2.rectangle(image, (x, y), (x + w, y + h), (222,228,251), -1)

cv2.imshow('image', image)
cv2.waitKey()
Run Code Online (Sandbox Code Playgroud)