raspberry piカメラでpython,opencvを用いて特定色の検出を行いたい

Question

下記のコードは赤色を検出するコードです。
検出する色を違う色に変えたいのですが、どのように書き換えたら良いでしょうか

import cv2
import numpy as np

def find_rect_of_target_color(image):
  hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV_FULL)
  h = hsv[:, :, 0]
  s = hsv[:, :, 1]
  v = hsv[:, :, 0]
  mask = np.zeros(h.shape, dtype=np.uint8)
  mask[((h < 50) | (h > 200)) & (s > 100)] = 255
  mask, contours, hierarchy = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
  rects = []
  for contour in contours:
    approx = cv2.convexHull(contour)
    rect = cv2.boundingRect(approx)
    rects.append(np.array(rect))
  return rects

capture = cv2.VideoCapture(0)
while cv2.waitKey(30) < 0:
    _, frame = capture.read()
    rects = find_rect_of_target_color(frame)
    if len(rects) > 0:
        rect = max(rects, key=(lambda x: x[2] * x[3]))
        cv2.rectangle(frame, tuple(rect[0:2]), tuple(rect[0:2] + rect[2:4]), (0, 0, 255), thickness=2)
    cv2.imshow('red', frame)
capture.release()
cv2.destroyAllWindows()

Answer 1

プログラムの前に、HSVの色空間で赤だけ(あるいは他の特定の色)を取り出すとはどういうことか、というのがわかっていないと話にならないのですが、そちらは大丈夫ですか?

Comments

[退会済みユーザー]

HSVのそれぞれの値を決めてその値の色の部分を画面上で四角く囲むようにすると考えていました。

[thkana]

H/S/Vのそれぞれの意味はわかっていますか?　赤をとりだす、というのはどういう値を選べばいいかとか。

[退会済みユーザー]

返信遅れてすみません。
Hは色相S彩度ｖは明度ということはわかります。赤だとHの範囲が255として230から20くらいでしょうか。

[thkana]

アナログ時代のテレビだとHue(色合い)Chroma(色の濃さ)Brightness(明るさ)という調整があって(それぞれHue,Saturation, Valueに対応)、それをいじってもらうと感覚が掴めたのですが...

色相環というくらいで、Hueについては範囲が255として255と0が繋がっている、というのもいいですね?
で、今のプログラムは、Hueが200を超えるか50未満でSaturationが100を超える部分を赤と判断している(Pythonの詳細を知らなくてもこのコードを見ればそれくらいの想像は出来るでしょう)のですから、そこの判断基準をあなたの期待する範囲に変えてやればいいです。
別に間違えてもパソコンが爆発したりはしませんから、値を変えて試してみたらいかがでしょう。ちゃんと知りたいのなら、Pythonの入門書の一冊ぐらいはこなす必要があるかも知れませんが。

Answer 2

赤、青、緑を識別するサンプルコードが記載されています。
https://qiita.com/seigot/items/008c95306dbd99a07309
ご参考まで。

Answer 3

こちらで対象にしたい色の境界値を調べて、
https://www.peko-step.com/tool/hsvrgb.html

下記のしきい値の判定を変更するとできると思います。

mask[((h < 50) | (h > 200)) & (s > 100)] = 255

Comments

[退会済みユーザー]

さっきの質問は変な質問をしたとおもったのでなしでお願いします。
新しい質問なのですが、「= 255」はなにを表しているのでしょうか。

[fana]

自分で書いたコードの意味を訊くんですか？