光流法二值化图像识别代码‘

以下是使用光流法进行二值化图像识别的 Python 代码：

import cv2

cap = cv2.VideoCapture(0) # 打开摄像头

_, frame1 = cap.read() # 读取第一帧
frame1_gray = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY) # 转为灰度图像

# 设置参数
lk_params = dict(winSize=(15, 15), maxLevel=4, criteria=(cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 0.03))

while True:
    _, frame2 = cap.read() # 读取当前帧
    frame2_gray = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY) # 转为灰度图像
    
    # 计算光流
    flow = cv2.calcOpticalFlowFarneback(frame1_gray, frame2_gray, None, 0.5, 3, 15, 3, 5, 1.2, 0)
    magnitude, angle = cv2.cartToPolar(flow[...,0], flow[...,1])
    
    # 二值化图像识别
    threshold = 10 # 阈值
    mask = cv2.threshold(magnitude, threshold, 255, cv2.THRESH_BINARY)[1] # 二值化
    contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE) # 查找轮廓
    for contour in contours:
        area = cv2.contourArea(contour)
        if area > 100: # 过滤面积小于 100 的轮廓
            cv2.drawContours(frame2, [contour], -1, (0, 255, 0), 2) # 绘制矩形
            
    cv2.imshow('frame', frame2) # 显示图像
    
    # 更新帧
    frame1_gray = frame2_gray.copy()
    
    if cv2.waitKey(1) & 0xFF == ord('q'): # 按下 'q' 键退出循环
        break

cap.release() # 释放摄像头
cv2.destroyAllWindows() # 关闭所有窗口

代码中的 `calcOpticalFlowFarneback` 函数用于计算光流，`threshold` 变量为阈值，`cv2.threshold` 函数用于将灰度图像二值化，`cv2.findContours` 函数用于查找轮廓，`cv2.drawContours` 函数用于绘制矩形。