python实现摄像头实时球体检测

宋恺

python实现摄像头实时球体检测

此程序需要安装opencv
可以直接在cmd中输入：

1. pip install opencv-python

复制代码

但因为服务器在国外，所以会很慢，建议在cmd中输入：

1. python -m pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

复制代码

一劳永逸的办法：
https://fishc.com.cn/thread-143635-1-1.html

                               
登录/注册后可看大图

代码如下：

1. import numpy as np
   
2. import cv2
   
3. 
   
4. cap = cv2.VideoCapture(0)
   
5. ret = cap.set(3, 640)  # 设置帧宽
   
6. ret = cap.set(4, 480)  # 设置帧高
   
7. font = cv2.FONT_HERSHEY_SIMPLEX  # 设置字体样式
   
8. kernel = np.ones((5, 5), np.uint8)  # 卷积核
   
9. 
   
10. if cap.isOpened() is True:  # 检查摄像头是否正常启动
    
11.     while(True):
    
12.         ret, frame = cap.read()
    
13.         gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)  # 转换为灰色通道
    
14.         hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)  # 转换为HSV空间
    
15. 
    
16.         lower_green = np.array([35, 50, 100])  # 设定绿色的阈值下限
    
17.         upper_green = np.array([77, 255, 255])  # 设定绿色的阈值上限
    
18.         #  消除噪声
    
19.         mask = cv2.inRange(hsv, lower_green, upper_green)  # 设定掩膜取值范围
    
20.         opening = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)  # 形态学开运算
    
21.         bila = cv2.bilateralFilter(mask, 10, 200, 200)  # 双边滤波消除噪声
    
22.         edges = cv2.Canny(opening, 50, 100)  # 边缘识别
    
23.         # 识别圆形
    
24.         circles = cv2.HoughCircles(
    
25.             edges, cv2.HOUGH_GRADIENT, 1, 100, param1=100, param2=10, minRadius=10, maxRadius=500)
    
26.         if circles is not None:  # 如果识别出圆
    
27.             for circle in circles[0]:
    
28.                 #  获取圆的坐标与半径
    
29.                 x = int(circle[0])
    
30.                 y = int(circle[1])
    
31.                 r = int(circle[2])
    
32.                 cv2.circle(frame, (x, y), r, (0, 0, 255), 3)  # 标记圆
    
33.                 cv2.circle(frame, (x, y), 3, (255, 255, 0), -1)  # 标记圆心
    
34.                 text = 'x:  '+str(x)+' y:  '+str(y)
    
35.                 cv2.putText(frame, text, (10, 30), font, 1, (0, 255, 0), 2, cv2.LINE_AA, 0)  # 显示圆心位置
    
36.         else:
    
37.             # 如果识别不出，显示圆心不存在
    
38.             cv2.putText(frame, 'x: None y: None', (10, 30), font, 1, (0, 255, 0), 2, cv2.LINE_AA, 0)
    
39.         cv2.imshow('frame', frame)
    
40.         cv2.imshow('mask', mask)
    
41.         cv2.imshow('edges', edges)
    
42.         k = cv2.waitKey(5) & 0xFF
    
43.         if k == 27:
    
44.             break
    
45.     cap.release()
    
46.     cv2.destroyAllWindows()
    
47. else:
    
48.     print('摄像头异常!')
    

复制代码