Как заменить цвет пикселей cv2?

Question

[maryaTurova]

nparr = np.frombuffer(base64.b64decode(base64_data), np.uint8)
bgr = cv2.imdecode(nparr, cv2.IMREAD_GRAYSCALE)

im = ImageOps.grayscale(img)

colour = {}

#получаю словарь ("цвет":"кол-во")
for pixel in im.getdata():
	if pixel in colour:
		colour[pixel] += 1
	else:
		colour[pixel] = 1
		
if len(colour) > 2:
	#получаю 3 макс.значения в виде кортежа
	aux = sorted([(v,k) for (k,v) in colour.items()])
	res = [(v,k) for (k,v) in aux[-3:]]

[(197, 121), (13, 179), (0, 1936)]

Как в "bgr"с помощью cv2 заменить первое значение на последнее, т.е. пиксели с 197 заменить на пиксели 0 ?

Answer 1

[Vindicar]

В cv2 изображения - это массивы numpy, так что все трюки с индексацией из numpy работают. Очень советую их изучить.

find = 1,2,3 # какие значения заменять
replace = 4,5,6 # на какие значения заменять
mask = (bgr == find)[..., 0]
bgr[mask] = replace

Comments

[maryaTurova]

Спасибо!
Я уже так пробовал, но цвет не меняется.

from PIL import Image, ImageOps
import numpy as np
import cv2
image='image.png'
img = Image.open(image)
im = ImageOps.grayscale(img)
bgr = cv2.imread(image, cv2.IMREAD_GRAYSCALE)
cv2.imshow('img',bgr)
cv2.waitKey(0)
colour={}
for pixel in im.getdata():
    if pixel in colour:
        colour[pixel] += 1
    else:
        colour[pixel] = 1
if len(colour) > 2:
    aux = sorted([(v,k) for (k,v) in colour.items()])
    res = [(v,k) for (k,v) in aux[-3:]]
    print(res,res[0][0],res[2][0])
    find = res[0][0] # какие значения заменять
    replace = res[2][0] # на какие значения заменять
    mask = (bgr == find)[..., 0]
    bgr[mask] = replace
    cv2.imshow('img',bgr)
    cv2.waitKey(0)

Пробовал даже все пиксели которые в кортеже есть .....без изменений.
В градации серого имеется тень, я вот и хочу от нее избавится

[maryaTurova]

могут ли пиксели в grayscale у PIL и cv2 отличатся?

[Vindicar]

maryaTurova, хороший вопрос - вполне возможно. К слову, а зачем тебе вообще PIL? сv2 умеет загрузать изображения самостоятельно.

[maryaTurova]

Та я не пойму как с помощью cv2 реализовать похожее getdata() от PIL. т.е. создать словарь (цвет: количество пикселей)

[Vindicar]

maryaTurova, Тебе нужно посчитать по каждому цвету (BGR) или по каналам отдельно? Потому что по цвету есть такой приём со Stackoverflow.

import numpy as np
arr = bgr
# Выстраиваем пиксели в ряд.
arr = arr.reshape(-1, arr.shape[-1])
# Сортируем по "возрастанию" цвета, чтобы одинаковые пиксели шли подряд
sorted_arr = arr[np.lexsort(arr.T), :]
# Используем разницу между соседями, чтобы понять, где заканчивается "цепочка" одинаковых пикселей
diff_idx = np.where(np.any(np.diff(sorted_arr, axis=0), 1))[0]
# Получаем набор уникальных пикселей
unique_rows = [sorted_arr[i] for i in diff_idx] + [sorted_arr[-1]]
# считаем, сколько раз встретился каждый уникальный пиксель
counts = np.diff(
    np.append(np.insert(diff_idx, 0, -1), sorted_arr.shape[0] - 1))
# строим словарь вида "кортеж с цветом - количество".
counter = dict(zip(map(tuple, unique_rows), counts))

[U235U235]

maryaTurova, воспользуйтесь cv2.calcHist()

[Vindicar]

U235U235, я не припомню, как ведёт себя calcHist() при работе с несколькими каналами? Он строит гистограмму комбинаций каналов, или же отдельную гистограмму по каждому каналу?

[U235U235]

Vindicar, строит, судя по документации, по каждому каналу. Но в данном случае это неважно, т.к. у ТС im - одноканальная картинка.