画像処理についてわからないです

Question

与えられた画像の信号機の色を赤色から緑色に変換するプログラムを書きたくて、エラーメッセージが出たが、何故ダメなのかわからないので教えていただきたいです。

import cv2
import matplotlib.pyplot as plt

img2 = cv2.imread( "traffic_light.png" )
img3 = cv2.cvtColor( img2, cv2.COLOR_BGR2RGB )

for y in img3[110:160]:
  for x in img3[170:220]:
    if img3[y,x,0]>=210:
      img3[y,x,0]=0 
      img3[y,x,1]=255

plt.imshow(img3)

エラーメッセージ

    14 for y in img3[110:160]:
     15   for x in img3[170:220]:
---> 16     if img3[y,x,0]>=210:
     17       img3[y,x,0]=0
     18       img3[y,x,1]=255

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

画像はこのような画像です

Comments

[tiitoi]

質問に元の画像も添付できますか？

Answer 1

forループのインデックスに指定するのは、指定範囲のx座標やy座標に相当する整数値で、img3のスライスではありません。

-for y in img3[110:160]:
+for y in range(110, 160):
-  for x in img3[170:220]:
+  for x in range(170, 220):