как ускорить поиск цвета пикселя изображения, которого нет в данном списке цветов

Question

Цель: я хотел бы ускорить процесс поиска значения пикселя изображения, не содержащего один из цветов в данной таблице цветов RGB, и сопоставить их с другим изображением с суффиксом _mistakes.png.

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

import glob
import numpy as np
import os
import cv2
import os.path
# the given list of defined RGB colors.
CLASSES = {
        0: [0, 0, 0],
        1:[255, 0, 0],
        2:[0, 0, 255],
        3:[0, 255, 0], 
        4:[50, 255, 50],
        5:[100, 255, 100]
        }
for image_path in glob.glob("*.png"):
    name = os.path.split(image_path)[-1]
    _name = os.path.splitext(name)[0]
    img = cv2.imread(image_path)
    img_height, img_width, _ = img.shape
    img_mistakes = np.zeros((img.shape))

    color_codes = np.array(list(CLASSES.values()))

    # the following two for loops take so long.
    for row in range(img_height):
        for col in range(img_width):
            if not (img[row,col] == color_codes).all(1).any():
                img_mistakes[row, col] = [200, 200, 200]  # a chosen color

cv2.imwrite(_name + '_mistakes' + '.png', img_mistakes)

Answer 1

Возможно, есть даже более быстрый путь, чем этот, но это начало!Мои деньги на @divakar, чтобы узнать это - подсказка, подсказка; -)

#!/usr/local/bin/python3
import numpy as np
import cv2

# Open image into numpy array
im=cv2.imread('start.png')

# Work out how one pixel of each colour we are looking for looks
black = [0,0,0]
blue  = [255,0,0]
red   = [0,0,255]
green = [0,255,0]

# Find all pixels where the 3 RGB values match the sought colour
blacks = np.all(im==black, axis=2)
blues  = np.all(im==blue , axis=2)
reds   = np.all(im==red  , axis=2)
greens = np.all(im==green, axis=2)

# Make empty (black) output array same size as input image
mistakes = np.zeros_like(im)
# Make anything not matching any of our colours into [200,200,200]
mistakes[~(blacks | blues | reds | greens)] = [200,200,200]

# Save result
cv2.imwrite("result.png",mistakes) 

start.png

Результат: