18834161486 发表于 2024-2-23 11:17

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 发表于 2024-2-23 12:47

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

18834161486 发表于 2024-2-23 13:56

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

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

名字全被注册啦 发表于 2024-2-23 19:12

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

18834161486 发表于 2024-2-24 08:18

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

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

lvtaode0657 发表于 2024-2-26 20:28

都是牛人,学习了

ccber 发表于 2024-3-8 22:26

以前读大学的时候,利用opencv做了一个项目,就是差分法获取轮廓,再snake模型贴近。
页: [1]
查看完整版本: python通过帧差法实现运动目标检测