python 行ごとに画像の画素ごとのRGBの個数を出したい

Question

pythonで画像を読み込み、その画像の各行ごと、画素のRGB値がそれぞれ何個あるのかを知りたいです。
以下のような画像を読み込んだ時に以下のような結果が理想です。

１行目　cleam 9
２行目　red 1, white3, yello2, black3
・
・
・

自分で作る時にまずは行ごとではなく全体の画素ごとのRGB値がどの数あるのかを調べるために以下のプログラムを作りました。

from PIL import Image
from numpy import *
import numpy as np
#np.set_printoptions(threshold=np.inf)

#画像の読み込み
im = Image.open('22_resize.png')

#各色ごとの画素の数を数える変数
black = 0
white = 0
red = 0
cleam = 0
blue = 0
violet = 0

for i in im.getdata():
    if i[0]<=50 and i[1]<=50 and i[2]<=50:
       black += 1
    if i[0]>=200 and i[1]>=200 and i[2]>=200:
       white += 1
    if i[0]>200 and i[1]<50 and i[2]<50:
        red += 1
    if i[0]>100 and i[1]>100 and i[2]>90:
       cleam += 1
    if i[0]<100 and i[1]<50 and i[2]>200:
        blue += 1
    if i[0]>100 and i[1]<50 and i[2]>90:
        violet += 1

print ('black=' + str(black))
print ('white='+str(white))
print ('red='+str(red))
print ('cleam='+str(cleam))
print ('blue='+str(blue))
print ('violet='+str(violet))

上のプログラムをイジって行ごとの数値を出そうとしたのですがどのように進めていけばいいのかで止まってしまいました。
どなたかわかる方のお力を貸して欲しいです。

Answer 1

以下の手順でできます。

• 1. 画像の画素値 (R, G, B) を色の名前 (質問の if 文で定義されているもの) に変換する。

→ 色ごとに関数を適用するのに numpy.apply_along_axis を使う。

• 1. 画像を行ごとに取得する。

for row in range(len(img)):
    img(row)

• 1. その行の色の種類及び数を数える。

collections.Countor を使う。

from PIL import Image
import numpy as np


def rgb_to_name(rgb):
    r, b, g = rgb
    if r <= 50 and b <= 50 and g <= 50:
        return "BLACK"
    if r >= 200 and b >= 200 and g >= 200:
        return "WHITE"
    if r > 200 and b < 50 and g < 50:
        return "RED"
    if r > 100 and b > 100 and g > 90:
        return "CLEAM"
    if r < 100 and b < 50 and g > 200:
        return "BLUE"
    if r > 100 and b < 50 and g > 90:
        return "VIOLET"

    return "OTHER"


img = np.array(Image.open("tttt.jpeg"))
img = np.apply_along_axis(rgb_to_name, 2, img)

from collections import Counter

for row in range(len(img)):
    cnt = Counter(img[row])
    print(f"row: {row}, count: {dict(cnt)}")

row: 0, count: {'WHITE': 289}
row: 1, count: {'WHITE': 289}
row: 2, count: {'WHITE': 289}
row: 3, count: {'WHITE': 289}
row: 4, count: {'WHITE': 289}
row: 5, count: {'WHITE': 289}
row: 6, count: {'WHITE': 289}
row: 7, count: {'WHITE': 289}
row: 8, count: {'WHITE': 280, 'CLEAM': 9}
row: 9, count: {'WHITE': 20, 'CLEAM': 269}

Comments

[退会済みユーザー]

a