python,opencvにおけるCb成分画像作成について

Question

前提・実現したいこと

python初心者ですが画像処理に関するプログラムを書いています。
現在ファイル内の.jpg画像を全てCb成分の画像に変換するコードを書きたいです。

発生している問題・エラーメッセージ

しかし処理した画像が真っ白のものができてしまい画像の処理ができません。

該当のソースコード

import cv2
import numpy as np
import glob

gazousu = len(glob.glob('*.jpg'))

def numpy_gray(src):
    r, g, b = src[:, :, 0], src[:, :, 1], src[:, :, 2]
    gray = -0.1687 * r - 0.3313 * g + 0.5000 * b + 128

if __name__ == "__main__":

  count = 0
  files = glob.glob('*.jpg') 
  for f in files:

    count += 1
    count_padded = '%05d' % count

    w = 0
    while w < gazousu:

     img = cv2.imread(f)
     
     gray_num = numpy_gray(img)
    
     w = w + 1   
     write_file_name = count_padded + ".jpg"
     cv2.imwrite(write_file_name, gray_num)

試したこと

最後の３行を

w = w + 1
write_file_name = count_padded + ".jpg"
cv2.imwrite("gray_num.jpg", gray_num)

とすればCb成分の画像を生成できましたが、無論上書き保存の形になり最後の１枚しか生成できません。

どなたか助けていただけないでしょうか。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

numpy_gray()関数が何も値を返していないので
下記のように直してみてはいかがでしょうか。

def numpy_gray(src):
    r, g, b = src[:, :, 0], src[:, :, 1], src[:, :, 2]
    gray = -0.1687 * r - 0.3313 * g + 0.5000 * b + 128
    src[:, :, 0], src[:, :, 1], src[:, :, 2] = gray, gray, gray
    return src

Comments

[imariyaki]

ベストアンサーの方のおかげで解決できました。
今後の同様の質問者のために完成したコードを記載しておきます。
ありがとうございました

Answer 2

import cv2
import numpy as np
import glob

gazousu = len(glob.glob('*.jpg'))

def numpy_gray(src):
    r, g, b = src[:, :, 0], src[:, :, 1], src[:, :, 2]
    gray = -0.1687 * r - 0.3313 * g + 0.5000 * b + 128
    src[:, :, 0], src[:, :, 1], src[:, :, 2] = gray, gray, gray
    return src

    #   Cr画像
    #   gray = 0.500 * r - 0.4186 * g - 0.0813 * b + 128
    #   Cb画像
    #   gray = -0.1687 * r - 0.3313 * g + 0.5000 * b + 128

if __name__ == "__main__":

  count = 0
  files = glob.glob('*.jpg') 
  for f in files:

    count += 1
    count_padded = '%05d' % count

    w = 0
    while w < gazousu:

     img = cv2.imread(f)
     
     gray_num = numpy_gray(img)
     
     w = w + 1
     write_file_name = count_padded + ".jpg"
     cv2.imwrite(write_file_name, gray_num)  

Comments

[bsdfan]

opencvでは、RGBがBGRの順に入ってると思うのですが、それは大丈夫ですか？