pythonで多次元配列の要素の値を変更したい

Question

pythonで
画像をhsv変換→v値をある値以上の場合（今回は225）であれば値を変換→画像をそのv値を用いて再作成
を行いたいと思っております。下がそのコードです。

import os
import cv2
import numpy as np
from matplotlib import pyplot as plt
DATADIR = "./kaizen/"
category = "n"
path = os.path.join(DATADIR, category)
for i,image_name in enumerate(os.listdir(path)):
    img = cv2.imread(os.path.join(path, image_name)) 
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) 
    h,s,v= cv2.split(hsv) 
    v_replace = [225 if i > 225 else i for i in v]
    
    hsv = cv2.merge((h_replace,s,v))
    rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
    save_path='./M1/./kaizen/./h/_%d.jpg' %i
    cv2.imwrite(save_path, rgb)
コード

実行してみると下記のようなエラーコードが出ます。
おそらくv値が82×47の多次元配列になっていることが原因だと思うのですが，
どのようなコードに変更すればよいのでしょうか，，，なにか参考サイトや改善方法を教えて頂けると幸いです。

ValueError                                Traceback (most recent call last)
<ipython-input-7-fff9ea7a778d> in <module>
     11     h,s,v= cv2.split(hsv)
     12     print(h)
---> 13     h_replace = [225 if i > 225 else i for i in h]
     14 
     15     hsv = cv2.merge((h_replace,s,v))

<ipython-input-7-fff9ea7a778d> in <listcomp>(.0)
     11     h,s,v= cv2.split(hsv)
     12     print(h)
---> 13     h_replace = [225 if i > 225 else i for i in h]
     14 
     15     hsv = cv2.merge((h_replace,s,v))

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

Answer 1

単純にv[v>225] = 225でよいと思います。

import cv2
import numpy as np

img = cv2.imread('test.png') 
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
h,s,v= cv2.split(hsv)

v[v>225] = 225
hsv = cv2.merge((h,s,v))

rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
cv2.imwrite('ret.png', rgb)

Comments

[reonald]

ありがとうございます。
その関数を見落としていました，，，