python通过帧差法实现运动目标检测

18834161486

[Python] 纯文本查看 复制代码

import cv2
import numpy as np
import pyautogui

# 初始化背景帧

while True:
    # 背景图
    background = pyautogui.screenshot()
    background = cv2.cvtColor(np.array(background), cv2.COLOR_RGB2BGR)
    background = cv2.cvtColor(background, cv2.COLOR_BGR2GRAY)

    # 下一帧图像
    screenshot = pyautogui.screenshot()
    frame = cv2.cvtColor(np.array(screenshot), cv2.COLOR_RGB2BGR)
    gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 计算帧差
    diff = cv2.absdiff(background, gray_frame)

    # 阈值处理
    _, thresh = cv2.threshold(diff, 30, 255, cv2.THRESH_BINARY)

    # 形态学处理
    kernel = np.ones((5, 5), np.uint8)
    thresh = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel)

    # 查找轮廓
    contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    for contour in contours:
        x, y, w, h = cv2.boundingRect(contour)
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

    cv2.imshow('Frame', frame)

    key = cv2.waitKey(1) & 0xFF
    if key == ord('q'):
        break

cv2.destroyAllWindows()

Yifan2007

为啥不直接比色就行了，截两张图取点色

18834161486

Yifan2007 发表于 2024-2-23 12:47
为啥不直接比色就行了，截两张图取点色

比色也得截两张图呀，当前颜色得截图吧，对比的颜色得截图吧

名字全被注册啦

帧间差分法还是限制比较多的，背景最好是不动的，基本就限制了相机固定和背景固定。另外可视化出来的结果也很吃腐蚀膨胀的参数

18834161486

名字全被注册啦 发表于 2024-2-23 19:12
帧间差分法还是限制比较多的，背景最好是不动的，基本就限制了相机固定和背景固定。另外可视化出来的结果也 ...

是的背景不变效果还是可以的，我测试用的是电脑桌面上拖动图标

lvtaode0657

都是牛人，学习了

ccber

以前读大学的时候，利用opencv做了一个项目，就是差分法获取轮廓，再snake模型贴近。