python - 基于空白分割图像

Question

我试图将这个图像分成相等的部分，其中气泡行之间的“空白”所在，然后将所有这些子图像排列在一个长图像中，以便问题垂直排序。如何使用python以这种方式分割图像？我这样做是为了使用 OpenCV 对气泡表进行分级。请记住，我是 python 新手（但不是编码），所以如果你能解释每个代码块的目的是什么会很好。

Answer 1

这是在 Python/OpenCV 中执行此操作的一种方法。

• 读取输入
• 边框颜色上的颜色阈值作为蒙版
• 使用遮罩使边框变白
• 稍微扩大蒙版以包括黑色边框线
• 将该结果转换为灰度
• 大津阈值
• 应用形态学打开将文本连接到列状区域并反转
• 获取列表中的外部轮廓及其边界框
• 计算所有盒子的最大宽度
• 循环每个框，用填充裁剪并与前一个框堆叠
• 保存结果

输入：

import cv2
import numpy as np

# read input image
img = cv2.imread('abcd_test.png')

# define border color
lower = (0, 80, 110)
upper = (0, 120, 150)

# threshold on border color
mask = cv2.inRange(img, lower, upper)

# dilate threshold
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (15, 15))
mask = cv2.morphologyEx(mask, cv2.MORPH_DILATE, kernel)

# recolor border to white
img[mask==255] = (255,255,255)

# convert img to grayscale
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# otsu threshold
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU )[1] 

# apply morphology open
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (17,17))
morph = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel)
morph = 255 - morph

# find contours and bounding boxes
bboxes = []
bboxes_img = img.copy()
contours = cv2.findContours(morph, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
contours = contours[0] if len(contours) == 2 else contours[1]
for cntr in contours:
    x,y,w,h = cv2.boundingRect(cntr)
    cv2.rectangle(bboxes_img, (x, y), (x+w, y+h), (0, 0, 255), 1)
    bboxes.append((x,y,w,h))

# get largest width of bboxes
maxwidth = max(bboxes)[2]

# sort bboxes on x coordinate
def takeFirst(elem):
    return elem[0]

bboxes.sort(key=takeFirst)

# stack cropped boxes with 10 pixels padding all around
result = np.full((1,maxwidth+20,3), (255,255,255), dtype=np.uint8)
for bbox in bboxes:
    (x,y,w,h) = bbox
    crop = img[y-10:y+h+10, x-10:x+maxwidth+10]
    result = np.vstack((result, crop))

# save result
cv2.imwrite("abcd_test_mask.jpg", mask)
cv2.imwrite("abcd_test_white_border.jpg", img)
cv2.imwrite("abcd_test_thresh.jpg", thresh)
cv2.imwrite("abcd_test_morph.jpg", morph)
cv2.imwrite("abcd_test_bboxes.jpg", bboxes_img)
cv2.imwrite("abcd_test_column_stack.png", result)

# show images
cv2.imshow("mask", mask)
cv2.imshow("img", img)
cv2.imshow("thresh", thresh)
cv2.imshow("morph", morph)
cv2.imshow("bboxes_img", bboxes_img)
cv2.imshow("result", result)
cv2.waitKey(0)

边框蒙版图像：

带边框的图像变为白色：

阈值和形态图像：

边界框图像：

裁剪和堆叠的列图像：