Отслеживание белого шара на белом фоне (Python / OpenCV)

Я использую OpenCV в Python 3, чтобы обнаружить белый / черный шар на белом поле и дать его точное (x, y, радиус) и цвет.

Я использую функции cv2.Canny () и cv2.findContours (), чтобы найти его, но проблема в том, что cv2.Canny () не всегда определяет полную форму круга (в большинстве случаев только 3/4 круга).
Поэтому, когда я использую cv2.findContours (), он не обнаруживает его как контур.

Посмотри пожалуйста :

изображение 1 а также изображение 2

Это самый важный код:

gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
blurred = cv2.GaussianBlur(gray, (5, 5), 0)
edged = cv2.Canny(blurred, 10, 40) # 10 and 40 to be more perceptive
contours_canny= cv2.findContours(edged.copy(), cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)[-2]

После этого я использую:
approx = cv2.approxPolyDP(contour,0.01*cv2.arcLength(contour,True),True)

Так что, если вы можете помочь мне найти решение, которое было бы здорово!
Может быть, есть функция, которая завершает форму круга, чтобы я мог его обнаружить?

3

Решение

Я хотел бы предложить, чтобы вы применили некоторую промежуточную фильтрацию цвета перед использованием хитрого обнаружения. Это гарантирует, что обнаружение острых краев будет происходить на четко определенной границе между изображением шара и полем.

Вот код Python, который использует трекбары, чтобы вы могли настроить цветовой порог:

cv2.namedWindow('temp')
cv2.createTrackbar('bl', 'temp', 0, 255, nothing)
cv2.createTrackbar('gl', 'temp', 0, 255, nothing)
cv2.createTrackbar('rl', 'temp', 0, 255, nothing)
cv2.createTrackbar('bh', 'temp', 255, 255, nothing)
cv2.createTrackbar('gh', 'temp', 255, 255, nothing)
cv2.createTrackbar('rh', 'temp', 255, 255, nothing)

bl_temp=cv2.getTrackbarPos('bl', 'temp')
gl_temp=cv2.getTrackbarPos('gl', 'temp')
rl_temp=cv2.getTrackbarPos('rl', 'temp')

bh_temp=cv2.getTrackbarPos('bh', 'temp')
gh_temp=cv2.getTrackbarPos('gh', 'temp')
rh_temp=cv2.getTrackbarPos('rh', 'temp')
thresh=cv2.inRange(frame,(bl_temp,gl_temp,rl_temp),(bh_temp,gh_temp,rh_temp))
2

Другие решения

Вы можете использовать cv.HoughCircles, чтобы оценить наиболее подходящий полный круг. При отслеживании мяча хорошо то, что шарик всегда будет изображаться в виде круга на изображении, независимо от угла наклона камеры.

1

По вопросам рекламы [email protected]