Как в opencv определить доминирующий цвет?

Question

[Михаил]

На вход дается изображение и нужно определить какой цвет - доминант.
или красный или зеленый или черный.
но вот как реализовать совсем не знаю..

Answer 1

[Сергей Горностаев]

За минуту нагуглил пример из документации и несколько статей - раз, два, три. Ещё минута понадобилась на то, чтобы проверить работу кода:

import cv2
import numpy as np
from sklearn.cluster import KMeans
from collections import Counter


def get_dominant_color(image, k=4):
    image = image.reshape((image.shape[0] * image.shape[1], 3))

    clt = KMeans(n_clusters = k)
    labels = clt.fit_predict(image)

    label_counts = Counter(labels)

    dominant_color = clt.cluster_centers_[label_counts.most_common(1)[0][0]]

    return list(dominant_color)


bgr_image = cv2.imread('image.png')
hsv_image = cv2.cvtColor(bgr_image, cv2.COLOR_BGR2HSV)
dom_color = get_dominant_color(hsv_image)
dom_color_hsv = np.full(bgr_image.shape, dom_color, dtype='uint8')
dom_color_bgr = cv2.cvtColor(dom_color_hsv, cv2.COLOR_HSV2BGR)
output_image = np.hstack((bgr_image, dom_color_bgr))
cv2.imshow('Dominant Color', output_image)
cv2.waitKey(0)

Comments

[Михаил]

Спасибо!

Answer 2

[Dimonchik]

ну задачка не одной командой делается, хотя бы потому что алгоритмы разные

https://stackoverflow.com/questions/28793985/find-...

https://habr.com/post/156045/
https://stackoverflow.com/questions/48480648/what-...

Comments

[Михаил]

Спасибо!

Answer 3

[Andrey Dugin]

Это легко делается с помощью PIL (Pillow) и numpy:

dominant_color = numpy.array(image.histogram(), dtype=np.int32).reshape(-1, 256).argmax(axis=1)

Comments

[eegmak]

что такое PIL (Pillow)?
в вашем примере можно выбрать прямоугольник для подсчета внутри него?