Настроить (изменить) цвета изображения (пикселя) - python - PullRequest
Новая
       61

Настроить (изменить) цвета изображения (пикселя) - python

1 голос
/ 13 апреля 2019

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

Пример:

Logo_1 logo_2

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

Я пытался: 

from PIL import Image
filename ="./Logo.jpg"

picture = Image.open(filename, 'r')
_colors = [(255, 255, 255), (128, 128, 0), (128, 128, 128), (192, 128, 0), (128, 64, 0), (0, 192, 0), (128, 64, 128), (255, 255, 255)]
width, height = picture.size

for x in range(0, width):
    for y in range(0, height):
        current_color = picture.getpixel((x,y))
        # print (current_color)
        if current_color in _colors:
            picture.putpixel((x,y), (255,5, 255))
            # print ("Y")

picture.save("./test/change.png")

Приведенный выше код является довольно распространенным кодом, который предлагается для большинства из них. Но это довольно сложно, так как он заменяет пиксель в списке " _colors " Выходное изображение:

wrong Output

Есть ли решение вышеуказанной проблемы? Любой умный способ справиться с этим, используя машинное обучение? Любое решение с использованием другого языка программирования?

1 Ответ

2 голосов
/ 13 апреля 2019

Я не знаком с PIL, и я слышал, что это медленно. Итак, вот версия OpenCV: 

# for red color, it's easier to work with negative image 
# since hue is in [170, 180] or [0,10]
hsv_inv = cv2.cvtColor(255-img, cv2.COLOR_BGR2HSV)

# these are cyan limit, but we're working on negative image, so...
lower_range = np.array([80,0,0])
upper_range = np.array([100,255,255])

# mask the red
mask = cv2.inRange(hsv_inv, lower_range, upper_range)

# replace red by green
green_hsv = hsv_inv.copy()
green_hsv[np.where(mask)] += np.array([60,0,0], dtype=np.uint8)    
green_img = 255 - cv2.cvtColor(green_hsv, cv2.COLOR_HSV2BGR)

purple_hsv = hsv_inv.copy()
purple_hsv[np.where(mask)] -= np.array([30,0,0], dtype=np.uint8)
purple_img = 255 - cv2.cvtColor(purple_hsv, cv2.COLOR_HSV2BGR)

И результат, пожалуйста, игнорируйте галочки, как я показал их через matplotlib.

green

purple

...