python—openCV实时追踪乒乓球

"""
张耀
2021年08月20日
"""
import cv2
import numpy as np


cap = cv2.VideoCapture(0)
while cap.isOpened():
    # 获取摄像头图像
    r, img = cap.read()
    # 截获乒乓球颜色
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    miny = np.array([5, 100, 250])
    maxy = np.array([25, 255, 255])
    # 乒乓球掩膜
    mask = cv2.inRange(hsv, miny, maxy)
    yellow = cv2.bitwise_and(img, img, mask=mask)
    # 图像处理
    gray_y = cv2.cvtColor(yellow, cv2.COLOR_BGR2GRAY)
    r, threading_y = cv2.threshold(gray_y, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
    k = np.ones((10, 10), np.uint8)
    threading_y = cv2.morphologyEx(threading_y, cv2.MORPH_CLOSE, k, iterations=2)
    # 获取轮廓
    r, counter, h = cv2.findContours(threading_y, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    n = len(counter)
    if n != 0:
        loc = []
        ri = []
        for i in range(n):
            (x, y), r = cv2.minEnclosingCircle(counter[i])
            loc.append((int(x), int(y)))
            ri.append(int(r))
        # 筛选轮廓
        dex = ri.index(max(ri))
        # 描绘轮廓
        cv2.circle(img, loc[dex], ri[dex], (0, 0, 255), 2)
    cv2.imshow('i', img)
    c = cv2.waitKey(1)
    if c == 27:
        break
cap.release()
cv2.destroyAllWindows()