17

我使用了一种自适应阈值技术来创建如下图所示的图片：

在此处输入图像描述

我使用的代码是：

image = cv2.adaptiveThreshold(image, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 45, 0)

然后，我使用此代码获取轮廓：

cnt = cv2.findContours(image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)[0]

我的目标是使用外轮廓内的所有像素生成蒙版，因此我想将对象内的所有像素填充为白色。我怎样才能做到这一点？

我已尝试使用下面的代码创建蒙版，但生成的蒙版似乎与应用自适应阈值后的图像没有什么不同

mask = np.zeros(image.shape[:2], np.uint8)
cv2.drawContours(mask, cnt, -1, 255, -1)
4

2 回答 2

20

你所拥有的几乎是正确的。如果你看一下你的阈值图像，它不起作用的原因是你的鞋子对象在图像中有间隙。具体来说，您所追求的是您希望鞋子的周边是全部连接的。如果发生这种情况，那么如果您提取最外部的轮廓（这是您的代码正在执行的操作），您应该只有一个代表对象外周长的轮廓。填充轮廓后，您的鞋子应该是完全坚固的。

由于鞋子的周边不完整且破损，这会导致白色区域断开连接。如果您使用findContours查找所有轮廓，它只会找到每个白色形状的轮廓，而不是最外周。因此，如果您尝试使用findContours，它会为您提供与原始图像相同的结果，因为您只需找到图像内每个白色区域的周长，然后用 填充这些区域findContours。

您需要做的是确保图像完全关闭。我建议您做的是使用形态学将所有断开连接的区域一起关闭，然后findContours对这个新图像运行调用。具体来说，执行二进制形态关闭。这样做的目的是，它需要靠近在一起的断开的白色区域并确保它们是连接的。使用形态闭合，或者使用类似 7 x 7 方形结构元素的东西来闭合鞋子。您可以将此结构化元素视为白色区域之间的最小间隔，以将它们视为连接。

因此，请执行以下操作：

import numpy as np
import cv2 
image = cv2.imread('...') # Load your image in here
# Your code to threshold
image = cv2.adaptiveThreshold(image, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 45, 0)    

# Perform morphology
se = np.ones((7,7), dtype='uint8')
image_close = cv2.morphologyEx(image, cv2.MORPH_CLOSE, se)

# Your code now applied to the closed image
cnt = cv2.findContours(image_close, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)[0]
mask = np.zeros(image.shape[:2], np.uint8)
cv2.drawContours(mask, cnt, -1, 255, -1)

此代码实质上采用您的阈值图像，并将形态关闭应用于该图像。之后，我们找到这个图像的外部轮廓，并用白色填充它们。FWIW，我下载了你的阈值图像，并自己尝试了这个。这就是我从你的图像中得到的：

在此处输入图像描述

于 2014-12-04T18:56:47.610 回答
2

一种简单的方法是关闭前景中的孔以形成单个轮廓，cv2.morphologyEx()其中cv2.MORPH_CLOSE

在此处输入图像描述

现在外轮廓已填充，我们可以找到外轮廓，cv2.findContours()并使用cv2.fillPoly()白色填充所有像素

在此处输入图像描述

import cv2

# Load in image, convert to grayscale, and threshold
image = cv2.imread('1.png')
gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)[1]

# Close contour
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (7,7))
close = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel, iterations=1)

# Find outer contour and fill with white
cnts = cv2.findContours(close, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]
cv2.fillPoly(close, cnts, [255,255,255])

cv2.imshow('close', close)
cv2.waitKey()
于 2019-10-02T22:59:07.770 回答