使用opencv將與特定顏色不匹配的顏色更改為白色

我正在嘗試使用 opencv 從影像中獲取特定元素。到目前為止，我所做的是減少影像中的顏色數量，并獲得了一個顏色字典及其計數。我現在想要做的是將所有與此特定顏色不匹配的顏色替換為白色，以便我可以迭代影像并創建不同的影像，其中每個影像只有一個該特定顏色的物件。

這是我到目前為止所做的：

def showOnlyOneColor(img, rgb_key):
    print("Getting only one color")
    rgb_key = rgb_key.split("-")
    r = int(rgb_key[0])
    g = int(rgb_key[1])
    b = int(rgb_key[2])

    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    rgb_low = np.array([r,g,b])
    rgb_high = np.array([r,g,b])
    mask = cv2.inRange(hsv, rgb_low, rgb_high)
    contours = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    contours = contours[0] if len(contours) == 2 else contours[1]
    cv2.fillPoly(mask, contours, (255, 255, 255))
    result = cv2.bitwise_and(img,img,mask=mask)


    return result

第二次嘗試：

def showOnlyOneColor(img, rgb_key):
    print("Getting only one color")
    rgb_key = rgb_key.split("-")
    r = int(rgb_key[0])
    g = int(rgb_key[1])
    b = int(rgb_key[2])

    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    rgb_low = np.array([r,g,b])
    rgb_high = np.array([r,g,b])
    mask = cv2.inRange(hsv, rgb_low, rgb_high)

    img[mask!=255] = (255, 255, 255)

    return img

這導致所有迭代的白色影像

這是我用來減少影像中顏色數量的方法：

def reduceNumberOfColors(img):
    div = 128
    return img // div * div   div // 2

最新嘗試：

def showOnlyOneColor(img, rgb_key):
    print("Getting only one color")
    rgb_key = rgb_key.split("-")
    r = int(rgb_key[0])
    g = int(rgb_key[1])
    b = int(rgb_key[2])
    hsv_color = colorsys.rgb_to_hsv(r,g,b)

    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    rgb_low = np.array(hsv_color)
    rgb_high = np.array(hsv_color)
    mask = cv2.inRange(hsv, rgb_low, rgb_high)

    img[mask!=255] = (255, 255, 255)

    return img

結果影像之一顯示了一些東西，其余影像是白色的

我如何組織 rgb：

def getColorCount(img):
    color_dict = {}
    print("Getting color count")
    for i in tqdm(range(img.shape[0])):
        for j in range(img.shape[1]):
            color = img[i,j]
            r = color[0]
            g = color[1]
            b = color[2]
            rgb_askey = str(r) "-" str(g) "-" str(b)
            if rgb_askey not in color_dict.keys():
                color_dict[rgb_askey] = 1
            else:
                color_dict[rgb_askey]  = 1
    return color_dict

uj5u.com熱心網友回復：

這是在 Python/OpenCV 中執行所需操作的一種方法。注意顏色的順序是 B、G、R。

輸入：

import cv2
import numpy as np

# read image
img = cv2.imread('corn.jpg')

# do simple color reduction
imgcopy = img.copy()
div = 128
imgcopy = div * ( imgcopy // div )   div // 2

# get list of unique colors
list_bgr_colors = np.unique(imgcopy.reshape(-1, imgcopy.shape[2]), axis=0)
print(list_bgr_colors)
print(list_bgr_colors[1])

# save reduced color image
cv2.imwrite("corn_reduced_colors.png", imgcopy)

# display reduced color image
cv2.imshow("reduced_colors", imgcopy)
cv2.waitKey(0)   

# loop over colors in list and change all non-specified colors to white
i = 1
for color in list_bgr_colors:

    # threshold on the specified color
    lower=np.array((color))
    upper=np.array((color))
    mask = cv2.inRange(imgcopy, lower, upper)

    # change all non-specified color to white
    result = imgcopy.copy()
    result[mask!=255] = (255, 255, 255)

    # save results
    cv2.imwrite("corn_color_{0}.png".format(i), result)

    # display result
    cv2.imshow("result", result)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    
    # increment
    i  = 1

減少顏色串列：

[[ 64  64  64]
 [ 64  64 192]
 [ 64 192  64]
 [ 64 192 192]
 [192 192 192]]

縮小的彩色影像：

單獨的彩色影像：

uj5u.com熱心網友回復：

OpenCV 中的顏色順序為 B、G、R。

所以在 Python/OpenCV 中試試這個。

def showOnlyOneColor(img, bgr_key):
    print("Getting only one color")
    bgr_key = bgr_key.split("-")
    b = int(bgr_key[0])
    g = int(bgr_key[1])
    r = int(bgr_key[2])
    bgr_low = np.array((b,g,r))
    bgr_high = np.array((b,g,r))
    mask = cv2.inRange(img, bgr_low, bgr_high)

    img[mask!=255] = (255, 255, 255)

    return img

添加

我認為您的問題在于您假設在使用 colorsys 將 RGB 轉換為 HSV 時具有 RGB 顏色。你的顏色實際上是 B、G、R。所以要使用 colorsys，你需要先將你的影像轉換為 RGB，或者在使用 colorsys 之前更好地反轉你從 B、G、R 到 R、G、B 提取的顏色的順序。

你的字典顏色實際上是 B、G、R 而不是 R、G、B。所以在使用 colorsys 時反轉它們或稍后反轉它們。

uj5u.com熱心網友回復：

這在 Python/OpenCV 中對我有用。請注意，OpenCV 顏色的順序為 B、G、R。

import cv2
import numpy as np

# read image
img = cv2.imread('corn.jpg')

# threshold on yellow color
lower=(0,170,215)
upper=(70,255,255)
mask = cv2.inRange(img, lower, upper)

# change all non-yellow to white
result = img.copy()
result[mask!=255] = (255, 255, 255)

# save results
cv2.imwrite('corn_yellow.jpg',result)

# display result
cv2.imshow("mask", mask)
cv2.imshow("result", result)
cv2.waitKey(0)
cv2.destroyAllWindows()

標籤：Python 麻木的 opencv

上一篇：如何洗掉此影像的特定顏色？

下一篇：計算分割和混洗影像中的垂直邊緣坐標（位置）