Opencv 色 範囲

Question

Hough変換で、得られた円の中心座標から、円の中心の色を求めました。
それが、以下のコードです。

import numpy as np
import cv2

img = cv2.imread("c:/temp/test.jpg")

# グレースケールに変換する。
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
grey_canny = cv2.Canny(img, 500.0, 300.0)

# ハフ変換で円検出する。
circles = cv2.HoughCircles(
    grey_canny, cv2.HOUGH_GRADIENT, dp=1.0, minDist=28, param1=100, param2=9, minRadius=11,maxRadius=15
)

# 検出結果を描画する。
if circles is not None:
    circles = circles.squeeze(axis=0)  # (1, NumCircles, 3) -> (NumCircles, 3)

#円の中心座標と、色を表示する。    
    for cx, cy, r in circles.astype(int):
        print(f"{(cx, cy)} = {img[cy, cx]}")
    
#円の描画
    for cx, cy, r in circles:
        # 円の円周を描画する。
        cv2.circle(img, (cx, cy), r, (0, 255, 0), 2)
        # 円の中心を描画する。
        cv2.circle(img, (cx, cy), 2, (0, 255, 0), 2)
        


# 結果を保存する。
cv2.imwrite("c:/temp/test1.jpg", img)
cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

この円の色が、

hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
lower = np.array([30, 24, 0])    
upper = np.array([90, 255, 255])

のような範囲の時は、そのまま円を描画し、今範囲外の時は別の色で円を描画する。
もしくは、円を描画しないようなプログラムを作成したいのですが、どのようにすればよいでしょうか？

Answer 1

hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
bin_img = cv2.inRange(hsv, (30, 24, 0), (90, 255, 255))

で (30, 24, 0) <= (H, S, V) <= (90, 255, 255) は255, そうでない画素は0という2値画像を予め作成しておいて、ループの中で参照した画素が255でなければ、その色の範囲外ということで描画をスキップすればよいです。

import numpy as np
import cv2

img = cv2.imread("c:/temp/test.jpg")

# グレースケールに変換する。
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
grey_canny = cv2.Canny(img, 500.0, 300.0)

# ハフ変換で円検出する。
circles = cv2.HoughCircles(
    grey_canny,
    cv2.HOUGH_GRADIENT,
    dp=1.0,
    minDist=28,
    param1=100,
    param2=9,
    minRadius=11,
    maxRadius=15,
)

# (30, 24, 0) <= (H, S, V) <= (90, 255, 255) は255, そうでない画素は0という2値画像を作成する。
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
bin_img = cv2.inRange(hsv, (30, 24, 0), (90, 255, 255))

# 検出結果を描画する。
if circles is not None:
    circles = circles.squeeze(axis=0)  # (1, NumCircles, 3) -> (NumCircles, 3)

    # 円の描画
    for cx, cy, r in circles:
        if bin_img[cy, cx] == 0:
            continue  # 指定した範囲外の色の場合は描画をスキップする。

        # 円の円周を描画する。
        cv2.circle(img, (cx, cy), r, (0, 255, 0), 2)
        # 円の中心を描画する。
        cv2.circle(img, (cx, cy), 2, (0, 255, 0), 2)


# 結果を保存する。
cv2.imwrite("c:/temp/test1.jpg", img)
cv2.imshow("img", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

Comments

[lebdom_11]

非常に助かりました。
いつもわかりやすくありがとうございます。
少し質問なのですが、

for cx, cy, r in circles:
if bin_img[cy, cx] == 0:
continue # 指定した範囲外の色の場合は描画をスキップする。

このcontinueは、どういう意味でしょうか？
if文を繰り返すという意味ですか？
素人質問過ぎるかもしれませんが、すみません。

[tiitoi]

continue はループ内の最初に戻るという処理です。
これによって、continue 以下に書かれた処理がスキップされるので、範囲外の色の場合は描画されないということが実現できます。

[tiitoi]

プログラミングの文法の話なので、「Python continue」で調べてみてください

[lebdom_11]

なるほど！！ありがとうございます！
もう少し勉強してみます！！