Opencv 円 中心 色

Question

python,Opencvを用いてHough変換により円の中心座標を取得しました。
その中心の色を調べたいのですが、

color = img[centers]

IndexError: arrays used as indices must be of integer (or boolean) type

というエラーが出てしまいます。

どうすれば良いですか？

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:
        # 円の円周を描画する。
        cv2.circle(img, (cx, cy), r, (0, 255, 0), 2)
        # 円の中心を描画する。
        cv2.circle(img, (cx, cy), 2, (0, 255, 0), 2)


#円の中心座標を表示
centers = circles[:, :2]
print(centers)

#円の中心の色を表示
color = img[centers]
print(f"Image[{centers}] = BGRA{color}")





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

Answer 1

以下のようにすればいいです。

• 円は複数検出されることもある。
• astype(int) で int 型にキャストする。

- centers = circles[:, :2]
- print(centers)
- color = img[centers]
- print(f"Image[{centers}] = BGRA{color}")
+ for cx, cy, r in circles.squeeze(axis=0).astype(int):
+     print(f"{(cx, cy)} = {img[cy, cx]}")

追記

q_sane_q さんがご指摘くださってるように円の中心の色を取得する前に円の中心に色を塗ってしまっているので、取得された色は元の画素値ではなく、その色になってしまいます。
描画する画像はコピーをとっておいたほうがよいと思います。

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,
)

img_copy = img.copy()  # 描画用にコピーしておく。

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

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

for cx, cy, r in circles.squeeze(axis=0).astype(int):
    print(f"{(cx, cy)} = {img[cy, cx]}")


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

Comments

[q_sane_q]

ただこのコード、上でcirclesを上書きしてしまってるのと、
あと検出した円の上に(0, 255, 0)で上塗りしてしまってるのが要注意ポイントですね

[tiitoi]

たしかに質問のコードだと、取得した色がcv2.circle() で塗った色になってしまいますね。

[lebdom_11]

円の中心を描画しているものを消したら、ちゃんと円の中心の色を検出できました！
お二人ともありがとうございます。

Answer 2

「中心座標」がそのままでは整数ではないのでしょう．
適当に整数に丸めてやればよいのではないでしょうか．

Comments

[lebdom_11]

ありがとうございます！！