以下是使用OpenCV和Python编写的排水管排水报警代码示例。该代码通过比较连续帧之间的差异,检测排水管是否排泥,并在超过一分钟未检测到排泥时触发报警。

import cv2
import time

# 设置报警阈值
DIFF_THRESHOLD = 0.5  # 差异阈值
ALARM_DURATION = 60  # 报警持续时间(秒)

# 加载视频文件或摄像头
video_capture = cv2.VideoCapture(0)  # 如果要从视频文件中读取,请将参数设置为文件路径

# 初始化变量
last_frame = None  # 上一帧
last_alarm_time = None  # 上一次报警时间

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

    # 转换为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    gray = cv2.GaussianBlur(gray, (21, 21), 0)

    # 如果是第一帧,则跳过差异检测
    if last_frame is None:
        last_frame = gray
        continue

    # 计算当前帧与上一帧的差异
    frame_diff = cv2.absdiff(last_frame, gray)
    _, frame_diff = cv2.threshold(frame_diff, 25, 255, cv2.THRESH_BINARY)

    # 计算差异比例
    diff_ratio = cv2.countNonZero(frame_diff) / (frame.shape[0] * frame.shape[1])

    # 如果差异比例超过阈值
    if diff_ratio > DIFF_THRESHOLD:
        # 如果距离上一次报警时间超过一分钟,则触发报警
        if last_alarm_time is None or time.time() - last_alarm_time >= ALARM_DURATION:
            print("排泥报警!")
            last_alarm_time = time.time()

    # 显示当前帧和差异图像
    cv2.imshow("Frame", frame)
    cv2.imshow("Frame Diff", frame_diff)

    # 按下q键退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

    # 更新上一帧
    last_frame = gray

# 释放视频捕获对象和关闭窗口
video_capture.release()
cv2.destroyAllWindows()

请注意,此代码仅为示例,可能需要根据实际情况进行修改和调整


原文地址: https://www.cveoy.top/t/topic/hYzl 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录