[Python]OpenCVで色の割合を求める

Question

前提

素人なのでご教授いただければ幸いです。
Open-CVで色の割合を求めるシステムをつくっている。

実現したいこと

紅葉を定点観測して得た画像を分析して紅葉日と落葉日を判断させたい。その中で今回、撮影した画像のなかの色の割合を出したいと考えています。

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

---

以下のようなエラーが出ています。
NameError Traceback (most recent call last) ~\AppData\Local\Temp\ipykernel_30076\3627228303.py in <module> 16 17 # 出現回数(ピクセル数)をカウントする ---> 18 counter = collections.Counter(color_list) 19 20 # ピクセル数と色をpandasデータフレーム化 NameError: name 'collections' is not defined

該当のソースコード

以下のコードを作成しました。

import numpy as np
import cv2
from matplotlib import pyplot as plt
img = cv2.imread('pic-20221101T004942Z.png')
h, w, _ = img.shape

 ＃全画素の色リストを作成
 ＃色情報は B*1000000 + G*1000 + R に変換し
 ＃0埋めした9桁の文字列に　 
color_list = []
for y in range(h):
    for x in range(w):
        color_list.append(str(img[y, x, 0]*1000000 +
                          img[y, x, 1]*1000 + 
                          img[y, x, 2]).zfill(9))
        
 ＃出現回数(ピクセル数)をカウントする        
counter = collections.Counter(color_list)

 ＃ピクセル数と色をpandasデータフレーム化
df = pd.DataFrame(data=[list(counter.values()), list(counter.keys())],
                 index = ['pixels', 'BGR']).T

 ＃ピクセル数が全体の1%未満の行を削除(ノイズとみなす)
df2 = df[df['pixels'] > h*w*0.01]

 ＃降順に並べ替え
df3 = df2.sort_values(by=['pixels'], ascending=False)

 ＃ndarrayに変換し一行ずつ読み出しリストに追加
 ＃色情報は9桁の文字列なので3文字ずつ読み出し 
 ＃255で割って0～1の数字に変換
 ＃順番をBGR→RGBに変更
x_list =[]
rgb_list =[]
for i in df3.to_numpy():
    x_list.append(i[0])  
    rgb_list.append([int(i[1][6:9])/255, 
                     int(i[1][3:6])/255,
                     int(i[1][0:3])/255])

＃円グラフの作成
plt.pie(x_list, 
        colors=rgb_list, 
        wedgeprops={'linewidth': 1, 'edgecolor':'black'}, 
        counterclock=None, 
        startangle=90)
plt.show()

参考にしたサンプルコード

https://tat-pytone.hatenablog.com/entry/2022/04/25/203755
https://miseruit.com/2022/09/27/post-3447/

Comments

[can110]

カレントディレクトリに読込ファイル(～.png)はたしかに存在するでしょうか？

[KENJI]

はい、確かに存在していると認識しております。
現在はimg = cv2.imread('pic-20221101T004942Z.jpg')で行っており、
---------------------------------------------------------------------------
NameError Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_30076\3627228303.py in <module>
16
17 # 出現回数(ピクセル数)をカウントする
---> 18 counter = collections.Counter(color_list)
19
20 # ピクセル数と色をpandasデータフレーム化

NameError: name 'collections' is not defined

上記のようなエラーが出ております。

[can110]

エラーメッセージが異なるようですので質問本文を修正ください。
また、インデントがなくなっているのでソースコードはコードブロックで囲んでください。
なおcollectionsはimportしているでしょうか？

[jbpb0]

pythonのコードの一番最初の行のすぐ上に
```python
だけの行を追加してください
また、pythonのコードの一番最後の行のすぐ下に
```
だけの行を追加してください

または、
https://teratail.storage.googleapis.com/uploads/contributed_images/56957fe805d9d7befa7dba6a98676d2b.gif
を見て、そのようにしてみてください

現状、コードがとても読み辛いです
質問にコードを載せる際に上記をやってくれたら、他人がコードを読みやすくなり、コードの実行による現象確認もやりやすくなるので、回答されやすくなります

[KENJI]

can110様jbpb0様
このような仕様にできることを初めて知りました。
ありがとうございます。

Answer 1

Markdownを使用していないので、質問文にコピペした際に欠損したの
かもしれませんが、提示されたプログラムでは

# ピクセル数が全体の1%未満の行を削除(ノイズとみなす)
df2 = df[df['pixels'] > hw0.01]

と書かれていますが、コピペ元では

df2 = df[df['pixels'] > h*w*0.01]

と書いてあります。
上記を修正して、インデント付けたり足りないimportを追加
したりしたら、なんとなく動いた気がするのでご確認下さい。

ちなみに、コピペ元のサイトでは国旗のような色数が多くない
べた塗り画像で色の割合を求めていますが、実行してみた感じだと
色数が多い写真系の画像ではうまく行かないかもしれません。
プログラムの動作確認はコピペ元の国旗画像で行っていただくと
良いかと思います。

※ 「質問へのコメント」で、最初の質問と別のエラーが書いてありますが、
collectionsがないのは「import collections」してください

Comments

[KENJI]

何から何までありがとうございます。参考にさせていただき実施したところ
File "~AppData\Local\Temp\ipykernel_30076\4145440289.py", line 26
df2 = df[df['pixels'] > hw 0.01]
^
SyntaxError: invalid syntax

上記のようなエラーが出たのですが何が悪いかわかりますでしょうか…。
国旗画像に変更も行いました。

[poto568]

df2 = df[df['pixels'] > hw 0.01]

とすると、うちの環境でもSyntaxErrorになりますね。
「編集 2022/11/22 16:37」で記載されている質問文のpythonのソースコードでは
h と w と 0.01 の間にそれぞれ半角の＊が入っています。コメント欄の機能の問題なのか
現状では判断できませんので、念のためご確認下さい。

[KENJI]

入れてみると
---------------------------------------------------------------------------
NameError Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_30076\3831213342.py in <module>
20
21 # ピクセル数と色をpandasデータフレーム化
---> 22 df = pd.DataFrame(data=[list(counter.values()), list(counter.keys())],
23 index = ['pixels', 'BGR']).T
24

NameError: name 'pd' is not defined
上記のようなエラーがでました。
*をつけるとひとまずその部分に関してはエラーがなくなったように思えます。

[KENJI]

上記のようなエラーは何が原因だと考えられますでしょうか。
何度も申し訳ないのですがご教授いただければ幸いです。

[jbpb0]

> NameError: name 'pd' is not defined

質問者さんが参考にしてる
https://tat-pytone.hatenablog.com/entry/2022/04/25/203755
のコードには

import pandas as pd

がありますが、この質問の「該当のソースコード」にはありません

コードに上記を追加したら直ると思いますけど、なぜその行を含めてないのでしょうか？

【追記】
質問者さん
コードに「import pandas as pd」を追加しても、エラーが直らないのでしょうか？