画像の一部分の画素数を求めたい

Question

以前このような質問をしたものです。

リンク内容

このプログラムから黒、白それぞれ合計の画素数がいくつあるのかを求めたいです。
書き方を変えてfor文で数を数える方法などを自分で試しましたがなかなかうまくできません。

・読み込みたい画像

・出したい結果
cleam,red, yellow,blueに囲まれたwhiteの画素数　○個
clean,yellow,yellow,blueに囲まれたblackの画素数　○個

どなたか、わかるかたお願いいたします。

追記：以下のようなプログラムを追加しましたがうまくできませんでした。

for row in range(len(img)):
    cnt = Counter(img[row])

if i[0] < 50 and i[1] < 50 and i[2] < 50:#ここをRGBにしたら数えてくれる
    black += 1

    print(str(black))

Comments

[meg_]

ご自分で試行錯誤されたコードを掲載してください。（丸投げと思われると回答が付きにくくなりますよ）

[退会済みユーザー]

追記しました。

[surphy]

コードのインデントが間違っているので、それを直しましょう。

[退会済みユーザー]

インデントのミスを修正しました。このプログラムではif文にエラーが出てしまいました。配列の部分がおかしいことはわかるのですがRGB値を読み込む際にどのように求めたら良いかがわかりません。

[surphy]

インデントは少し改良されましたが、for文の中でif文の判定がしたいのでは？
今そうなっていないです。

またi[0]などの表記があり、なんとなく何したいかはわかりますが、iという配列やリストが出てきていないのでエラーになります。
修正してみてください。

[退会済みユーザー]

配列、リストの部分がまだ理解できていないです。
新しいfor文の中でrgbを定義していくのか、それとも新しく別の場所で定義するのか等ヒントもらえると嬉しいです。

自分で以下のようにrgbを定義して見ましたがprintされなかったです。
for row in range(len(img)):
cnt = Counter(img[row])
print(f"{row+1}, count: {dict(cnt)}")
def rgb_to_name(rgb):
r, b, g = rgb
if r < 50 and g < 50 and b < 50:#ここをRGBにしたら数えてくれる
black += 1
print(str(black))

[meg_]

以前の質問でベストアンサーに選ばれた回答のコードは理解されていますか？
今回はそのコードを追加修正したいとのことですので、そちらの理解が未だだと難しいかと思います。

Answer 1

PythonはC++などと比べると圧倒的に遅いので、可能な限りfor文を使うのを避けたいところです。

これが最速ではないと思いますが、numpyで一致する箇所を抽出して、BGRが一致する(=足し算の結果が3になる)ピクセルをカウントさせてはいかがでしょうか？

import cv2
import numpy as np
# OpenCVで読み込む
img = cv2.imread("img.png")

# BGRに分割する
img_b,img_g,img_r = cv2.split(img)

# グレースケールの値が一致する部分を1、他を0として返す関数
def color_mask(img_gray,val):
   img_ans = np.zeros_like(img_gray)
   img_ans[img_gray==val] = 1
   return img_ans

# BGRが全て255の箇所を抽出
img_white = color_mask(img_b,255) + color_mask(img_g,255) + color_mask(img_r,255)

# BGRが全て0の箇所を抽出
img_black = color_mask(img_b,0) + color_mask(img_g,0) + color_mask(img_r,0)

# 抽出した答えが3の箇所を回答
print("White: %s"%len(np.where(img_white.flatten()==3)[0]))
print("Black: %s"%len(np.where(img_black.flatten()==3)[0]))