使用OpenCV串行执行线段分割(裁剪)

时间:2018-12-28 17:28:26

标签: python opencv text-segmentation handwriting-recognition

我正在使用深度学习执行整页的离线手写识别。

主要思想是构建一个模型,该模型可以获取一行文本图像并为其提供相应的文本。对于此主要任务,是对页面中的每一行进行线分割并将其发送到模型。

但是,我通过略微修改here来应用以下代码。这里出现的主要问题是,它随机裁剪了图像的线条,我将其串行保存为segment_no_1,2,3 ....

当我(随机地)将这些分段的线传递给模型时,它无法生成相应的串行数字文本。

是否有合适的方法或算法可以像原始图像中那样用OpenCV串行执行线段分割。我已经在深度学习中找到了行细分,但是我不想使用它。

我的代码:

import cv2
import numpy as np
#import image
image = cv2.imread('input2.png')
#cv2.imshow('orig',image)
#cv2.waitKey(0)

#grayscale
gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
cv2.imshow('gray',gray)
cv2.waitKey(0)

#binary
ret,thresh = cv2.threshold(gray,127,255,cv2.THRESH_BINARY_INV)
cv2.imshow('second',thresh)
cv2.waitKey(0)

#dilation
kernel = np.ones((5,100), np.uint8)
img_dilation = cv2.dilate(thresh, kernel, iterations=1)
cv2.imshow('dilated',img_dilation)
cv2.waitKey(0)

#find contours
im2,ctrs, hier = cv2.findContours(img_dilation.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

#sort contours
sorted_ctrs = sorted(ctrs, key=lambda ctr: cv2.boundingRect(ctr)[0])

for i, ctr in enumerate(sorted_ctrs):
    # Get bounding box
    x, y, w, h = cv2.boundingRect(ctr)

    # Getting ROI
    roi = image[y:y+h, x:x+w]

    # show ROI
    cv2.imshow('segment no:'+str(i),roi)
    cv2.imwrite("segment_no_"+str(i)+".png",roi)
    cv2.rectangle(image,(x,y),( x + w, y + h ),(90,0,255),2)
    cv2.waitKey(0)

cv2.imwrite('final_bounded_box_image.png',image)
cv2.imshow('marked areas',image)
cv2.waitKey(0)

The final output of the code on IAM database form

segment_no_1.png作为第一线段可以从中间或有时倒数第二个找到,依此类推。

因此,应该进行什么修改才能找到与原始图像一样的(正确地)顺序正确的分段线。

我对代码的任何改进也受到高度赞赏。预先感谢。

1 个答案:

答案 0 :(得分:2)

我认为您应该遵循this,其中显示使用Python和OpenCV对轮廓进行排序。

我遵循的基本步骤是:

  1. 对图像进行模糊处理,如有必要,请先转换为灰度。
  2. 应用Canny边缘检测算法来查找每个字符的轮廓。
  3. 将边缘检测图像传递给自适应算法,该算法在考虑相邻点的情况下效果更好。
  4. 执行膨胀,以更好地进行线段分割。
  5. 在膨胀图像的副本上执行线分割,随机生成分段。
  6. 最后按“从上到下”的顺序对细分进行排序。