python - 在两行之间添加空间

我需要使用OpenCV或PIL在两行之间添加空间。

最佳答案

如果行的长度“足够”地变化，则以下方法可能有用:

使用 cv2.threshold 逆二值化图像。

使用 cv2.dilate 使用水平线内核将图像放大以强调线条。

使用 np.sum 将所有像素按行求和，并使用 np.diff 计算行之间的绝对差。

行之间的差异中将存在“步骤”，类似于行之间的步骤。设置阈值并使用 np.where 找到合适的索引。

使用 np.insert 将白线插入原始图像中找到的索引之前。在下面的示例中，索引是手动选择的。必须进行适当的自动化工作:将“步骤”排除在“背景”之外，在多行之间找到“步骤”。

这是一个代码片段:

import cv2
from matplotlib import pyplot as plt
import numpy as np
from skimage import io              # Only needed for web grabbing images, use cv2.imread for local images

# Read and binarize image
image = cv2.cvtColor(io.imread('https://i.stack.imgur.com/56g7s.jpg'), cv2.COLOR_RGB2GRAY)
_, image_bin = cv2.threshold(image, 128, 255, cv2.THRESH_BINARY_INV)

# Dilate rows by using horizontal line as kernel
kernel = np.ones((1, 51), np.uint8)
image_dil = cv2.dilate(image_bin, kernel)

# Sum pixels row-wise, and calculate absolute differences between the rows
row_sum = np.sum(image_dil / 255, 1, dtype=np.int32)
row_sum_diff = np.abs(np.diff(row_sum))

# Just for visualization: Summed row-wise pixels
plt.plot(row_sum)
plt.show()

# Find "steps" in the differences between the rows
step_thr = 100
step_idx = np.where(row_sum_diff > step_thr)[0]

# Insert n lines before desired index; simple hard-coding here, more work needs to be done for multiple lines
n_lines = 5
image_mod = np.insert(image, step_idx[1] + 1, 255 * np.ones((n_lines, image.shape[1]), np.uint8), axis=0)

# Result visualization
cv2.imshow('image', image)
cv2.imshow('image_dil', image_dil)
cv2.imshow('image_mod', image_mod)
cv2.waitKey(0)
cv2.destroyAllWindows()

膨胀的逆二值化图像:



可视化的“步骤”:



最终的输出用n = 5插入了白线:



如您所见，结果并不完美，但这是由于原始图像。在相应的行中，您具有第一行和第二行的部分。因此，不可能对这两者进行适当区分。可以在输出中添加很小的形态封闭，以消除这些伪像。

希望有帮助!

关于python - 在两行之间添加空间，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/59065161/