21 ноября 2016

Обнаружение кругов с OpenCV на Python

Для нахождения кругов в OpenCV имеется функция cv2.HoughCircles() она имеет множество аргументов, которые неплохо описаны в документации.
Перейдем непосредственно к коду:




import cv2
import numpy as np

#Получить изображение из файла
img = cv2.imread('pendulum.jpg',0)
#Фильтрация изображения
img = cv2.medianBlur(img,5)
cimg = cv2.cvtColor(img,cv2.COLOR_GRAY2BGR)

circles = cv2.HoughCircles(img,cv2.HOUGH_GRADIENT,1,20,                            param1=50,param2=30,minRadius=20,maxRadius=0)

circles = np.uint16(np.around(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('Обнаруженные круги',cimg)
cv2.waitKey(0)

cv2.destroyAllWindows()
 


Слева — исходное изображение, справа — с обнаруженными окружностями.

Заметим, что так как установлен параметр minRadius=20, то малая окружность не обнаружена, это можно использовать для фильтрации лишних окружностей.