使用opencv结合帧差法和背景减法 检测场景异常情况

一、帧差法检测异常

帧差法是一种简单的背景减法技术,用于检测当前帧和背景帧之间的差异。以下是使用OpenCV实现帧差法的Python代码示例:

import cv2

# 读取背景图像(背景应该是静止的)
background = cv2.imread('background.jpg', 0)

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

while True:
    # 读取当前帧
    ret, frame = cap.read()
    
    if not ret:
        break

    # 将当前帧转为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 计算当前帧与背景的差异
    diff = cv2.absdiff(gray, background)

    # 设置一个阈值,根据阈值判断差异区域
    _, threshold = cv2.threshold(diff, 30, 255, cv2.THRESH_BINARY)

    # 执行形态学操作,去除噪声
    kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
    threshold = cv2.morphologyEx(threshold, cv2.MORPH_OPEN, kernel)

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

    # 绘制检测到的轮廓
    for contour in contours:
        if cv2.contourArea(contour) > 1000:  # 设置一个面积阈值来排除小的轮廓
            x, y, w, h = cv2.boundingRect(contour)
            cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

    # 显示结果
    cv2.imshow('Frame', frame)

    if cv2.waitKey(1) & 0xFF == 27:  # 按Esc键退出
        break

cap.release()
cv2.destroyAllWindows()

用于检测是否下雨,漏水等情况。

二 截取摄像头3秒的时间,然后用帧差法

要截取摄像头的3秒时间并使用帧差法进行动态背景差异检测,你可以使用OpenCV库来完成这项任务。首先,你需要设置一个计时器,以便捕获3秒的视频。然后,你可以应用帧差法来检测背景变化。

import cv2

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

# 获取摄像头的帧速率
frame_rate = int(cap.get(5))

# 计时器(3秒)
duration = 3  # 3秒
frames_to_capture = frame_rate * duration

# 初始化背景
background = None

# 计数器
frame_count = 0

while True:
    ret, frame = cap.read()

    if not ret:
        break

    if frame_count < frames_to_capture:
        # 累积前景图像,以用于帧差法
        if background is None:
            background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        else:
            current_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
            cv2.accumulateWeighted(current_frame, background, 0.5)
            background = cv2.convertScaleAbs(background)

        frame_count += 1
    else:
        # 3秒时间结束,开始使用帧差法检测背景变化
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        diff = cv2.absdiff(background, gray)

        _, threshold = cv2.threshold(diff, 30, 255, cv2.THRESH_BINARY)

        kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
        threshold = cv2.morphologyEx(threshold, cv2.MORPH_OPEN, kernel)

        contours, _ = cv2.findContours(threshold, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

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

        cv2.imshow('Motion Detection', frame)

        if cv2.waitKey(1) & 0xFF == 27:
            break

cap.release()
cv2.destroyAllWindows()

这个代码会首先捕获3秒的视频作为背景,然后在3秒结束后应用帧差法来检测背景变化。检测到的背景变化会用绿色矩形框标记出来。你可以根据需要进行参数调整,以获取最佳的检测效果。

三、背景减法

背景减法 (Background Subtraction) 是一种常用于视频分析和物体跟踪的技术。它可以用来检测视频中的移动对象,并提取它们与背景的差异。以下是一个使用OpenCV库实现背景减法的Python示例代码:

import cv2

# 打开视频文件或摄像头
cap = cv2.VideoCapture('your_video.mp4')  # 替换为你的视频文件路径或0来使用摄像头

# 创建背景减法器
fgbg = cv2.createBackgroundSubtractorMOG2()

while True:
    ret, frame = cap.read()

    if not ret:
        break

    # 应用背景减法器
    fgmask = fgbg.apply(frame)

    # 反转前景掩码,以便提取前景对象
    fgmask = cv2.bitwise_not(fgmask)

    # 获取前景对象
    result = cv2.bitwise_and(frame, frame, mask=fgmask)

    cv2.imshow('Original', frame)
    cv2.imshow('Foreground', result)

    if cv2.waitKey(30) & 0xFF == 27:  # 按Esc键退出
        break

cap.release()
cv2.destroyAllWindows()

四、OpenCV 其他方法

OpenCV 中有几种方法可以检测静止画面中的异常情况,尤其是检测是否有异物进入视角。以下是一些常用的方法:

  1. 帧差法 (Frame Difference Method):这是一种简单的背景减法技术,通过比较当前帧和背景帧的差异来检测移动对象。如果差异超过某个阈值,就可以认为有异物进入视角。

  2. 背景减法 (Background Subtraction):这是一种更复杂的方法,它会建立一个背景模型,并检测与该模型不匹配的区域。如果某个区域在一段时间内发生变化,就可以检测到异常情况。

  3. 光流法 (Optical Flow):光流法可以检测移动的物体,并估计它们的速度和方向。如果光流的速度超过某个阈值,就可以认为有物体移动。

  4. 帧差法与背景减法的结合:结合使用帧差法和背景减法可以更好地检测静止画面中的异常情况。

  5. 运动检测算法 (Motion Detection Algorithms):OpenCV 提供了一些运动检测算法,如Mean-Shift和CamShift等,可以用于检测视频中的运动对象。这些算法通常用于跟踪运动物体,但也可用于异常检测。

你可以根据你的需求选择其中一种或多种方法来检测异常情况。需要根据具体情况来调整参数和阈值,以便得到最佳的检测效果。

五、结合帧差法和背景减法

结合帧差法和背景减法来检测摄像头3秒的视频中是否有变化是一种常见的方法。以下是一个示例代码,演示如何结合这两种技术来实现这个任务:

import cv2
import time

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

# 创建背景减法器
fgbg = cv2.createBackgroundSubtractorMOG2()

# 读取第一帧作为背景
ret, background = cap.read()

# 设置计时器,记录运行时间
start_time = time.time()
run_time = 0

while run_time < 3:  # 运行3秒钟
    ret, frame = cap.read()
    
    if not ret:
        break
    
    # 应用背景减法器
    fgmask = fgbg.apply(frame)

    # 反转前景掩码,以便提取前景对象
    fgmask = cv2.bitwise_not(fgmask)

    # 获取前景对象
    result = cv2.bitwise_and(frame, frame, mask=fgmask)

    # 计算帧差
    frame_diff = cv2.absdiff(frame, background)

    # 设置阈值,用于检测帧差
    threshold = 30
    _, thresh = cv2.threshold(cv2.cvtColor(frame_diff, cv2.COLOR_BGR2GRAY), threshold, 255, cv2.THRESH_BINARY)

    # 计算帧差中非零像素的数量
    nonzero_pixels = cv2.countNonZero(thresh)

    if nonzero_pixels > 100:  # 如果帧差中的非零像素数量超过阈值
        print("有变化检测到!")

    cv2.imshow('Original', frame)
    cv2.imshow('Foreground', result)

    if cv2.waitKey(30) & 0xFF == 27:  # 按Esc键退出
        break

    # 计算运行时间
    run_time = time.time() - start_time

cap.release()
cv2.destroyAllWindows()

在这个示例中,我们打开摄像头并创建了一个背景减法器。首先,我们读取第一帧作为背景。然后,我们循环处理摄像头捕获的每一帧,应用背景减法器和帧差法来检测是否有变化。如果检测到变化(非零像素数量超过阈值),则打印消息。

这个代码将运行3秒钟,然后退出。你可以根据需要调整帧差法和背景减法的参数,以适应不同的场景和需求。

猜你喜欢

转载自blog.csdn.net/March_A/article/details/133795314