Question

我有黑屏和白球的视频我想在每个帧中跟踪白球的中心（输出为x，y坐标）。所以我尝试使用cv2.HoughCircles，但无论我如何调整参数，我都不会100％跟踪。

所以我有另一个想法。因为当圆形的每个像素的平均值将获得圆的中心点。我决定尝试这个想法。但循环每个像素直到我们发现白色是不可行的。有没有办法裁剪（粗略）只获得白色圆圈像素？（例如，我的图像分辨率为1920x1080，但球的大小不会超过50x50像素。因此在python中处理它的速度太慢。那么有没有办法将图像中的区域裁剪为50x50？）

Answer 1

没关系，我已经明白了。这是我的解决方案

hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
lower_white = np.array([0,0,150], dtype=np.uint8)
upper_white = np.array([0,0,255], dtype=np.uint8)
mask = cv2.inRange(hsv, lower_white, upper_white)

indcies = np.nonzero(mask)
size = len(set(zip(indcies[0], indcies[1])))
avgX, avgY = 0, 0
minX, minY = 10000, 10000
maxX, maxY = -1, -1

for (x, y) in set(zip(indcies[0], indcies[1])):
    minX, minY, maxX, maxY = min(minX, x), min(minY, y), max(maxX, x), max(maxY, y)
    avgX += x
    avgY += y
avgX /= size
avgY /= size
radius = int(round(max(maxX - minX, maxY - minY)/2))
cv2.circle(image,(int(round(avgY)),int(round(avgX))),radius,(0,255,0),2)
plt.imshow(image)
plt.show()

参考：

How do I know the position of white areas detected using OpenCV in python?

如何在cv2 python

1 个答案: