基于视频图像的火灾检测算法2

检测要求：
1.视频帧满足R(x,y)>𝐺(𝑥,𝑦)>𝐵(𝑥,𝑦)、R(x,y)>190，𝐺(𝑥,𝑦)>100、𝐵(𝑥,𝑦)＜140、Y(x,y)≥Cr(x,y)、𝐶𝑟(x,y)≥Cb(x,y)、Y(x,y)>𝑌通道均值、Cr(x,y)>𝐶𝑟通道均值、Cb(x,y)<𝐶𝑏通道均值条件
2.满足利用动态特征火焰检测中帧差法相关系数的判断规则
3.满足把彩色图片利用gray_now=1.6R-G-0.8B转化为灰度图，根据灰度图利用最大类间方差法（OTSU）进行二值化，二值化的值为1
把满足以上条件的区域用红色的方框给框出来，并给出“Fire”的警告，不满足条件的给出“No Fire”的警告，展示处理后的视频，并保存下来

import cv2
import numpy as np
# 定义视频路径
video_path = "fire_detection.mp4"
# 定义视频输出路径
output_path = "output.avi"
# 定义视频帧大小
frame_width = 640
frame_height = 360
# 定义视频编码器
fourcc = cv2.VideoWriter_fourcc(*"MJPG")
# 定义视频输出对象
out = cv2.VideoWriter(output_path, fourcc, 20, (frame_width, frame_height))
# 定义灰度图转换系数
gray_now = 1.6
# 读取视频
cap = cv2.VideoCapture(video_path)
# 定义OTSU阈值
otsu_threshold = 0
# 定义帧差法阈值
frame_diff_threshold = 0.5
# 定义帧差法相关系数阈值
frame_corr_threshold = 0.8
# 定义计数器
count = 0
# 定义上一帧信息
last_binary = None
last_frame_diff_binary = None
while True:
    # 读取视频帧
    ret, frame = cap.read()
    if not ret:
        break
    # 调整视频帧大小
    frame = cv2.resize(frame, (frame_width, frame_height))
    # 将彩色图片转化为灰度图
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    gray = gray_now * frame[:, :, 2] - frame[:, :, 1] - 0.8 * frame[:, :, 0]
    gray = np.clip(gray, 0, 255).astype(np.uint8)
    # 进行OTSU二值化
    otsu_threshold, binary = cv2.threshold(gray, otsu_threshold, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
    # 定义蓝色、绿色、红色通道
    b, g, r = cv2.split(frame)
    # 计算通道均值
    mean_r = np.mean(r)
    mean_g = np.mean(g)
    mean_b = np.mean(b)
    # 计算Y、Cr、Cb通道
    Y = 0.299 * r + 0.587 * g + 0.114 * b
    Cr = r - Y
    Cb = b - Y
    # 满足条件的区域
    fire_area = (r > g) & (r > b) & (r > 190) & (g > 100) & (b < 140) & (Y >= Cr) & (Cr >= np.mean(Cr)) & (Cb <= np.mean(Cb)) & (Y > np.mean(Y))
    # 进行帧差法
    if count > 0:
        frame_diff = cv2.absdiff(binary, last_binary)
        frame_diff = cv2.medianBlur(frame_diff, 5)
        frame_diff_binary = np.zeros(frame_diff.shape, dtype=np.uint8)
        frame_diff_binary[frame_diff > frame_diff_threshold * 255] = 1
        if last_frame_diff_binary is not None:
            frame_corr = cv2.matchTemplate(frame_diff_binary, last_frame_diff_binary, cv2.TM_CCOEFF_NORMED)[0][0]
            if frame_corr > frame_corr_threshold:
                fire_area = fire_area | frame_diff_binary
        last_frame_diff_binary = frame_diff_binary.copy()
    # 绘制方框
    contours, _ = cv2.findContours(fire_area.astype(np.uint8), 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, 0, 255), 2)
        cv2.putText(frame, "Fire", (x, y - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
    # 输出视频帧
    out.write(frame)
    # 显示视频帧
    cv2.imshow("Fire Detection", frame)
    # 更新计数器和上一帧信息
    count += 1
    last_binary = binary.copy()
    # 按'q'键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()

代码2：在上述程序中，添加一段程序：只保留检测面积最大的火焰的目标

import cv2
import numpy as np
# 定义视频路径
video_path = "fire_detection.mp4"
# 定义视频输出路径
output_path = "output.avi"
# 定义视频帧大小
frame_width = 640
frame_height = 360
# 定义视频编码器
fourcc = cv2.VideoWriter_fourcc(*"MJPG")
# 定义视频输出对象
out = cv2.VideoWriter(output_path, fourcc, 20, (frame_width, frame_height))
# 定义灰度图转换系数
gray_now = 1.6
# 读取视频
cap = cv2.VideoCapture(video_path)
# 定义OTSU阈值
otsu_threshold = 0
# 定义帧差法阈值
frame_diff_threshold = 0.5
# 定义帧差法相关系数阈值
frame_corr_threshold = 0.8
# 定义计数器
count = 0
# 定义上一帧信息
last_binary = None
last_frame_diff_binary = None
while True:
    # 读取视频帧
    ret, frame = cap.read()
    if not ret:
        break
    # 调整视频帧大小
    frame = cv2.resize(frame, (frame_width, frame_height))
    # 将彩色图片转化为灰度图
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    gray = gray_now * frame[:, :, 2] - frame[:, :, 1] - 0.8 * frame[:, :, 0]
    gray = np.clip(gray, 0, 255).astype(np.uint8)
    # 进行OTSU二值化
    otsu_threshold, binary = cv2.threshold(gray, otsu_threshold, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
    # 定义蓝色、绿色、红色通道
    b, g, r = cv2.split(frame)
    # 计算通道均值
    mean_r = np.mean(r)
    mean_g = np.mean(g)
    mean_b = np.mean(b)
    # 计算Y、Cr、Cb通道
    Y = 0.299 * r + 0.587 * g + 0.114 * b
    Cr = r - Y
    Cb = b - Y
    # 满足条件的区域
    fire_area = (r > g) & (r > b) & (r > 190) & (g > 100) & (b < 140) & (Y >= Cr) & (Cr >= np.mean(Cr)) & (Cb <= np.mean(Cb)) & (Y > np.mean(Y))
    # 进行帧差法
    if count > 0:
        frame_diff = cv2.absdiff(binary, last_binary)
        frame_diff = cv2.medianBlur(frame_diff, 5)
        frame_diff_binary = np.zeros(frame_diff.shape, dtype=np.uint8)
        frame_diff_binary[frame_diff > frame_diff_threshold * 255] = 1
        if last_frame_diff_binary is not None:
            frame_corr = cv2.matchTemplate(frame_diff_binary, last_frame_diff_binary, cv2.TM_CCOEFF_NORMED)[0][0]
            if frame_corr > frame_corr_threshold:
                fire_area = fire_area | frame_diff_binary
        last_frame_diff_binary = frame_diff_binary.copy()
    # 绘制方框
    contours, _ = cv2.findContours(fire_area.astype(np.uint8), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    max_area = 0
    max_area_contour = None
    for contour in contours:
        area = cv2.contourArea(contour)
        if area > max_area:
            max_area = area
            max_area_contour = contour
    if max_area_contour is not None:
        x, y, w, h = cv2.boundingRect(max_area_contour)
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2)
        cv2.putText(frame, "Fire", (x, y - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
    # 输出视频帧
    out.write(frame)
    # 显示视频帧
    cv2.imshow("Fire Detection", frame)
    # 更新计数器和上一帧信息
    count += 1
    last_binary = binary.copy()
    # 按'q'键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()