python - 填充轮廓圆

Question

我有一组图像，其中圆圈绘制为白色轮廓。但是，我想用白色填充整个圆圈。什么是快速的方法呢？以下是图片示例:

我曾尝试使用嵌套循环来实现这一点，但这需要花费很多时间，而且我有大约 150 万张图像。以下是我的代码:

roundRobinIndex = 0
new_image = np.zeros((img_w, img_h))
for row in range(540):
    for column in range(800):
        if image[row,column] == 255:
            roundRobinIndex = (roundRobinIndex + 1) % 2
        if roundRobinIndex == 1:
            new_image[row, column] = 255

Answer 1

使用cv2.fillPoly()填充圆形轮廓

import cv2

image = cv2.imread('1.png', 0)
thresh = cv2.threshold(image, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)[1]
cnts = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]
cv2.fillPoly(image, cnts, [255,255,255])

cv2.imshow('image', image)
cv2.waitKey()

注意:由于输入图像已经是二值图像，可以移除 Otsu 的阈值以获得稍微更快的性能，您可以直接在灰度图像上找到轮廓