opencvでif文に輝度値の項目を入れたい

Question

import cv2
import numpy as np

img = cv2.imread("input.png")
y_size, x_size, c = img.shape

for y in range(y_size):
    for x in range(x_size):
        if img[y, x] = [255, 0, 0]:
            print("blue")
        elif img[y, x] = [0, 255, 0]:
            print("green")
        else:
            print("black")       

   if img[y, x] = [255, 0, 0]:
                 ^
SyntaxError: invalid syntax

opencvを使用して，1つ目のように各画素の色をプログラムを作成したいのですが2つ目のようなエラーが出ます．

おそらく，現在のコードはopencvの指定した画素の輝度値を変更するもののためエラーが出ているのだと思うのですが何か方法はないでしょうか？

ご回答よろしくお願いします．

Answer 1

ifでの比較は=ではなく==を使用しましょう。
またimg[y, x]はlistではなくnumpy.ndarrayなので、比較すると[True,True,True]といったように、それぞれの要素を比較したものをlistで返してします。
ですのでall()を使用して、全要素が一致したらTrueを返すようにするのが良いでしょう。

for y in range(y_size):
    for x in range(x_size):
        if all(img[y, x] == [255, 0, 0]):
            print("blue")
        elif all(img[y, x] == [0, 255, 0]):
            print("green")
        else:
            print("black")

Comments

[ruuburubijutuka]

ありがとうございます．
初歩的な問題でした．

[y.nakamura]

opencvを調べるとimg[y,x]がlistではなくndarrayだったので追記しました。