opencvでの円検出について

Question

前提・実現したいこと

ハフ変換を用いた円検出を行いたいです。
パラメータの設定がうまくできません。

以下、円検出を行うサンプル画像です

該当のソースコード

import cv2
import numpy as np

img = cv2.imread('C:/Users/ito/Anaconda3/envs/Sample/pic/testimg/IMG_0557.jpg', 0)
img = cv2.medianBlur(img, 5)
cimg = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)

circles = cv2.HoughCircles(img, cv2.HOUGH_GRADIENT, 1, 10, param1=500, param2=100, minRadius=0, maxRadius=0)

circles = np.uint16(np.arround(circles))
for i in circles[0,:]:
    cv2.circle(cimg, (i[0], i[1]), i[2], (0,255,0), 2)
    cv2.circle(cimg, (i[0], i[1]), 2, (0,0,255), 3)
    

cv2.imshow('detected circles', cimg)
cv2.waitKey(0)
cv2.destroyAllWindows
cv2.imwrite('C:/Users/ito/Anaconda3/envs/Sample/pic/result/Hough_sample01.jpg', cimg)

試したこと

パラメータの設定変更

補足情報（FW/ツールのバージョンなど）

windows7 professional
python3.6.5
opencv3.3.1

Answer 1

以下のパラメータを小さい値にしたら、検出できるようになりました。

param2 – 手法依存の 2 番目のパラメータ. CV_HOUGH_GRADIENT の場合は,円の中心を検出する際の投票数の閾値を表します.これが小さくなるほど,より多くの誤検出が起こる可能性があります.より多くの投票を獲得した円が,最初に出力されます.

リファレンス をみてください。

import cv2
import matplotlib.pyplot as plt
import numpy as np

img = cv2.imread('test.jpg')

# グレースケールに変換
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# ガウシアンフィルタ
gray = cv2.medianBlur(gray, 5)

plt.imshow(gray, cmap='gray')
plt.axis('off')
plt.show()

# ハフ変換による円検出
circles = cv2.HoughCircles(gray, cv2.HOUGH_GRADIENT, 1, 20,
                           param1=200, param2=50)
circles = np.squeeze(circles, axis=0)  # (1, N, 3) -> (N, 3)

# 描画する。
if circles is not None:
    for x, y, radius in np.rint(circles).astype(int):
        print('center: ({}, {}), radius: {}'.format(x, y, radius))
        cv2.circle(img, (x, y), radius, (0, 255, 0), 5)
        cv2.circle(img, (x, y), 2, (0, 0, 255), 5)
    
plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
plt.axis('off')
plt.show()

Comments

[Taka11]

ありがとうございます！
自分のほうでも試してみたところできました！