利用GRB颜色模型过滤条件进行火灾检测

      火灾检测的相关介绍参考我另一篇博客：火灾检测方法总结
      本文RGB过滤条件参考论文：Fire detection using statistical color model in video sequences，提取效果比一般的RGB颜色过滤提取效果好些，我进行了测试，提取的区域更加饱和，代码参考博客：Opencv + python火焰分割加速版进行改写，其中的GRB过滤条件为：
                                                            

import cv2 as cv
import numpy as np

def contrast_brightness_demo(image, c, b):  #其中c为对比度，b为每个像素加上的值（调节亮度）
    blank = np.zeros(image.shape, image.dtype)   #创建一张与原图像大小及通道数都相同的黑色图像
    dst = cv.addWeighted(image, c, blank, 1-c, b) #c为加权值，b为每个像素所加的像素值
    ret, dst = cv.threshold(dst, 25, 255, cv.THRESH_BINARY)
    return dst


capture = cv.VideoCapture("/home/yasin/house5.mp4")

while(True):
    ret, frame = capture.read()
    cv.imshow("frame", frame)
    B = frame[:, :, 0]
    G = frame[:, :, 1]
    R = frame[:, :, 2]
    R_mean = np.mean(R)
    val1 = G /(R + 1)
    val2 = B /(R + 1)
    val3 = B /(G + 1) 
    fireImg = np.array(np.where(R > R_mean, np.where(R >= G, np.where(G >= B, np.where(val1 >= 0.25, np.where(val1 <=0.65, np.where(val2 >= 0.05,np.where(val2 <=0.45,np.where(val3 >=0.2 ,np.where(val3 <= 0.6, 255, 0), 0), 0), 0), 0), 0), 0), 0), 0))
    
    gray_fireImg = np.zeros([fireImg.shape[0], fireImg.shape[1], 1], np.uint8)
    gray_fireImg[:, :, 0] = fireImg
    gray_fireImg = cv.GaussianBlur(gray_fireImg, (7, 7), 0)
    gray_fireImg = contrast_brightness_demo(gray_fireImg, 5.0, 25)
    kernel = cv.getStructuringElement(cv.MORPH_RECT, (5, 5))
    gray_fireImg = cv.morphologyEx(gray_fireImg, cv.MORPH_CLOSE, kernel)
    dst = cv.bitwise_and(frame, frame, mask=gray_fireImg)
    cv.namedWindow("fire",0)
    cv.imshow("fire", dst)
    cv.namedWindow("gray_fireImg",0)
    cv.imshow("gray_fireImg", gray_fireImg)
    c = cv.waitKey(40)
    if c == 27:
        break

原始图：
                                  

测试结果：