分析圖像中的主色時刪除背景色（python，OpenCV）

Question

我正在使用https://www.pyimagesearch.com/2014/05/26/opencv-python-k-me-color-clustering/中的出色代碼（這不是我的！）。

我要解決的是，在繪制顏色時要刪除某些顏色-在這種情況下，我要在繪制直方圖時去除/不測量玩具的白色背景。 在其他情況下，當使用透明背景時，它會拉動灰色/白色選中的背景（例如https://i.stack.imgur.com/WLjsK.png ），因此我想同時移動這兩種“顏色”。

有人可以請我朝着解決這個問題的正確方法前進嗎？

非常感謝！！

import cv2
import numpy as np
from sklearn import metrics
import matplotlib.pyplot as plt

!wget "https://www.beatrix-potter-shop.co.uk/app/uploads/2017/07/PO1424_Benjamin.jpg"

# Load the image
image = cv2.imread("PO1424_Benjamin.jpg")
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) 

# Resize it for efficiency (color ratios are the same)
h, w, _ = image.shape
w_new = int(100 * w / max(w, h) )
h_new = int(100 * h / max(w, h) )

image = cv2.resize(image, (w_new, h_new));

# Step 1: Clustering colours with K-Means
from sklearn.cluster import KMeans

# Reshape the image to be a list of pixels
image_array = image.reshape((image.shape[0] * image.shape[1], 3))

# Clusters the pixels
clt = KMeans(n_clusters = 7)
clt.fit(image_array)

# By Adrian Rosebrock
def centroid_histogram(clt):
 # grab the number of different clusters and create a histogram
 # based on the number of pixels assigned to each cluster
 numLabels = np.arange(0, len(np.unique(clt.labels_)) + 1)
 (hist, _) = np.histogram(clt.labels_, bins = numLabels)

 # normalize the histogram, such that it sums to one
 hist = hist.astype("float")
 hist /= hist.sum()

 # return the histogram
 return hist

# Finds how many pixels are in each cluster
hist = centroid_histogram(clt)

# Sort the clusters according to how many pixel they have
zipped = list(zip(hist, clt.cluster_centers_))
zipped.sort(reverse=True, key=lambda x : x[0])
hist, clt.cluster_centers = zip(*zipped)

def plot_colors(hist, centroids):
    # initialize the bar chart representing the relative frequency
    # of each of the colors
    bar = np.zeros((50, 300, 3), dtype = "uint8")
    startX = 0

    # loop over the percentage of each cluster and the color of
    # each cluster
    for (percent, color) in zip(hist, centroids):
        # plot the relative percentage of each cluster
        endX = startX + (percent * 300)
        cv2.rectangle(bar, (int(startX), 0), (int(endX), 50),
            color.astype("uint8").tolist(), -1)
        startX = endX

    # return the bar chart
    return bar

# build a histogram of clusters and then create a figure
# representing the number of pixels labeled to each color
hist = centroid_histogram(clt)
bar = plot_colors(hist, clt.cluster_centers_)

# show our color bart
plt.figure()
plt.axis("on")
plt.imshow(bar)
plt.show()