python利用opencv简单识别红绿灯

#先装包和环境
import cv2
from PIL import Image
import numpy as np

#导入视频并自定义
cor_x, cor_y = -1, -1#定义自变量
fourcc = cv2.VideoWriter_fourcc(*'XVID')  # 使用XVID编码器
camera = cv2.VideoCapture('红绿灯.mp4')  # 从文件读取视频,需要修改成自己的文件的路径即可进行测试
fps = camera.get(cv2.CAP_PROP_FPS)  # 获取视频帧率
print('视频帧率：%d fps' % fps)
cor = np.array([[1, 1]])  # 初始值并无意义，只是为了能够使用np.row_stack函数

 

#创建坐标函数，并且赋予变量
def OnMouseAction(event, x, y, flags, param):#创建回调函数（even:指触发阶跃信号，点击鼠标返回实时坐标。
    global cor_x, cor_y, cor#将变量设置为全局变量.使其在下面函数外也能调用
    if event == cv2.EVENT_LBUTTONDOWN:
        print("左键点击")
        print("%s" % x, y)
        cor_x, cor_y = x, y#鼠标点击的x,y坐标赋予cor_x,cor_y
        cor_m = [cor_x, cor_y]
        cor = np.row_stack((cor, cor_m))#将鼠标返回的坐标进行矩形合并，并赋予全局变量
    elif event == cv2.EVENT_LBUTTONUP:
        cv2.line(img, (cor_x, cor_y), (cor_x, cor_y), (255,