找到轮廓上的主导色

2024-04-20 02:05:56 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图找到轮廓内的主导色(黑色或白色)。 我正在使用OpenCV读取图像并提取黑白图像。到目前为止我得到的是: enter image description here

绿色轮廓是轮廓,蓝色线条是边界框。所以在这个例子中,我试图提取数字87575220,但正如你所见,它也能识别出一些随机的伪影,比如字母G。我认为解决方法是找到轮廓内部的主导色,并且颜色应该接近白色。我不知道怎么做。在

我现在的代码是:

import argparse
import cv2
import imutils
import numpy as np

parser = argparse.ArgumentParser()
parser.add_argument("--image", "-i", required=True, help="Image to detect blobs from")
args = vars(parser.parse_args())

image = cv2.imread(args["image"])
image = imutils.resize(image, width=1200)
grey = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
(minVal, maxVal, minLoc, maxLoc) = cv2.minMaxLoc(grey)
maxval_10 = maxVal * 0.5
ret, threshold = cv2.threshold(grey, maxval_10, 255, cv2.THRESH_BINARY)
canny = cv2.Canny(grey, 200, 250)
lines = cv2.HoughLines(canny, 1, np.pi / 180, 140)

print(maxVal)

theta_min = 60 * np.pi / 180.
theta_max = 120 * np.pi / 180.0
theta_avr = 0
theta_deg = 0
filteredLines = []
for rho, theta in lines[0]:
    a = np.cos(theta)
    b = np.sin(theta)
    x0 = a * rho
    y0 = b * rho
    x1 = int(x0 + 1000 * (-b))
    y1 = int(y0 + 1000 * (a))
    x2 = int(x0 - 1000 * (-b))
    y2 = int(y0 - 1000 * (a))

    cv2.line(image, (x1, y1), (x2, y2), (0, 0, 255), 2)

    if theta_min <= theta <= theta_max:
        filteredLines.append(theta)
        theta_avr += theta

if len(filteredLines) > 0:
    theta_avr /= len(filteredLines)
    theta_deg = (theta_avr / np.pi * 180) - 90
else:
    print("Failed to detect skew")

image = imutils.rotate(image, theta_deg)
canny = imutils.rotate(canny, theta_deg)

im2, contours, hierarchy = cv2.findContours(canny, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
# cv2.drawContours(image, contours, -1, (0, 255, 0), 1)
cv2.imshow('Contours', im2)

boundingBoxes = []
filteredContours = []

for cnt in contours:
    (x, y, w, h) = cv2.boundingRect(cnt)
    if (h > 20 and h < 90 and w > 5 and w < h):
        if cv2.contourArea(cnt, True) <= 0:
            boundingBoxes.append((x, y, w, h))
            filteredContours.append(cnt)

for x, y, w, h in boundingBoxes:
    cv2.rectangle(image, (x, y), (x + w, y + h), (255, 0, 0), 2)

cv2.drawContours(image, filteredContours, -1, (0, 255, 0), 1)

cv2.imshow('Image', image)
cv2.imshow('Edges', canny)
cv2.imshow('Threshold', threshold)

cv2.waitKey(0)
cv2.destroyAllWindows()

这是原图: enter image description here


Tags: imageimportifnppicv2intgrey
3条回答

可以从每个轮廓创建遮罩:

mask = np.zeros(image.shape, dtype="uint8")
cv2.drawContours(mask, [cnt], -1, 255, -1)

然后计算遮罩内所有像素的平均值:

^{pr2}$

然后检查mean是否足够接近白色

由于颜色空间属性,颜色和平均值不匹配。我会创建一个柱状图,并选择最常见的一个(也可以应用一些颜色下降的采样)

在我开始搜索数字之前,我会尝试做一个投资回报率。你没有给出原始图像,所以这个例子是用你发布的图像制作的(盒子和轮廓都准备好了)。也应该和原著一起工作。步骤用示例代码编写。希望有帮助。干杯!在

示例代码:

import cv2
import numpy as np

# Read the image and make a copy then create a blank mask
img = cv2.imread('dominant.jpg')
img2 = img.copy()
h,w = img.shape[:2]
mask = np.zeros((h,w), np.uint8)

# Transform to gray colorspace and perform histogram equalization
gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)
equ = cv2.equalizeHist(gray)

# Transform all pixels above thershold to white
black = np.where(equ>10)
img2[black[0], black[1], :] = [255, 255, 255]

# Transform to gray colorspace and make a thershold then dilate the thershold
gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)
_, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
kernel = np.ones((15,15),np.uint8)
dilation = cv2.dilate(thresh,kernel,iterations = 1)

# Search for contours and select the biggest one and draw it on mask
_, contours, hierarchy = cv2.findContours(dilation,cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)
cnt = max(contours, key=cv2.contourArea)
cv2.drawContours(mask, [cnt], 0, 255, -1)

# Perform a bitwise operation
res = cv2.bitwise_and(img, img, mask=mask)

# Display the ROI
cv2.imshow('img', res)

结果:

enter image description here

相关问题 更多 >