【python-openCV】カラー画像における指定位置の色を判定する方法！

この記事ではpython&openCVを使って、カラー画像内における指定してピクセル位置での色を１２色相環の中から判定するサンプルプログラムをご紹介します。

ざっくりとイメージ図を紹介すると以下のような感じですね。

何に使えるかはわかりませんが、画像処理やpythonの勉強にはなると思います。

とりあえずやっていきましょう。笑

必要なライブラリを準備
色を判定する方法
指定位置における色判定サンプルプログラム
サンプルコードの実行結果
おわりに

必要なライブラリを準備

今回もいつもおなじみの以下２つのライブラリを使用します。

〇openCV

〇numpy

どちらもよく使うライブラリですので、この機会にインストールしておきましょう。

色を判定する方法

サンプルコードの紹介に入る前にどうやって色を判定するのかを簡単にご紹介しておきます。

今回は以下の流れで判定していきます。

①読み込んだ画像をまずHSVの色空間に変換

②指定ピクセル位置でのH（色相）を算出

③②のH（色相）と一番近いH（色相）を持つ色を１２色相環の中から判定

④判定された色を結果として出力する

HSVやら12色相環やら、よくわからないワードを並んでいますね。

１２色相環についてはおそらく中学校の美術の授業でならっているはずなので、検索すれば思い出せると思います

HSVについては以下の記事で詳細解説していますので、ぜひこちらもあわせてご覧ください。

【python-openCV】HSV色空間の概念を画像を使って詳細解説！

また、指定したいピクセル位置がどこなのかを簡単に確認する方法は以下の記事でご紹介しています。

興味があればこちらもあわせてご覧ください。

【python-openCV】クリック位置の座標を画像内に表示する方法

指定位置における色判定サンプルプログラム

だいたいの作業内容がわかったところで、さっそく冒頭に紹介した作業を実施してくれるサンプルプログラムをご紹介します。

import cv2
import numpy as np

#情報を入力する
file_name='sample.jpg'
search_posx=100
search_posy=100

#色を算出する関数
def color_judge(value):
    color_wheel=['red','red-orange','yellow-orange',
                 'yellow','yellow-grren','green',
                 'blue-green','green-blue','blue',
                 'blue-violet','violet','red-violet','red']
    index_value=int(round(value/(180/(len(color_wheel)-1)),0))
    return color_wheel[index_value]
    
#選択した場所の色を出力する
img=cv2.imread(file_name,cv2.IMREAD_COLOR)
cv2.circle(img,center=(search_posx,search_posy),radius=10,color=(255,255,255),thickness=3)
cv2.imwrite('img_point_'+str(search_posx)+'_'+str(search_posy)+'.jpg',img)
imghsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
color=color_judge(imghsv[search_posy,search_posx,0])
print(color)

#選択した場所の色を確認するための画像を出力
check=np.zeros((256,256,3),np.uint8)
check[:,:]=imghsv[search_posy,search_posx]
check = cv2.cvtColor(check, cv2.COLOR_HSV2BGR)
cv2.imwrite('img_check_'+str(search_posx)+'_'+str(search_posy)+'.jpg',check)

今回はプログラムの実行フォルダ内にあるsample.jpgという画像に対して処理を実行していくようにしています。

また、search_posx，search_posyが画像内の探索したい場所を指定しているところです。

プログラムを実行するとベースとなる画像(sample.jpg)に対して、指定位置に白丸を追加し、さらにその位置でのピクセル画像を拡大したものを出力するようにしています。

なお、今回のコードは１２色相環の中から算出しますが、世のなかには２４色相環なるものも存在しますので、より細かく色分けしたい場合は、コード中のcolor_wheelのところにデータを追加していけばＯＫです。