技術標籤：【Linux】jetson開發教程nvidialinuxqt

前言

不必害怕未知，無需恐懼犯錯，做一個Creator！

本文主要介紹jetson開發板中RGB判據火焰識別的實現。

一、原理流程圖

二、RGB判據原理

人眼的視網膜上有兩類感光器：錐狀體和桿狀體。其中，錐狀體對紅、綠、藍三種顏色的光很敏感，因此一般用於人眼觀看的顏色模型是RGB模型。
對於普通的火焰來說，它的紅色分量和綠色分量會很大，並且綠色分量會大於藍色分量，所以我們設下的簡單判據是：
S = (1-3.0*minValue/(R+G+B))
R > RT R>=G>=B S>=((255-R)*ST/RT)

三、實現程式碼及分析

int main(int argc, char *argv[])
{
   string filepath = "/home/cr/Desktop/QTfiles/FireTest/fire1.jpg";
   Mat inputImg = imread(filepath,1); //"1" means color images

    CheckColor(inputImg);
    return 0;
}

Mat CheckColor(Mat &inImg)
{
    Mat fireImg;
    fireImg.create(inImg.size(),CV_8UC1);

    int redThre = 115; // 115~135
    int saturationTh = 45; //55~65
    Mat multiRGB[3];
    int a = inImg.channels();
split(inImg,multiRGB); //將圖片拆分成R,G,B,三通道的顏色

    //每個畫素點進行RGB判據綜合判斷
    for (int i = 0; i < inImg.rows; i ++)
    {
        for (int j = 0; j < inImg.cols; j ++)
        {
            float B,G,R;
            B = multiRGB[0].at<uchar>(i,j); //每個畫素的R,G,B值
            G = multiRGB[1].at<uchar>(i,j);
            R = multiRGB[2].at<uchar>(i,j);

            /*B = inImg.at<uchar>(i,inImg.channels()*j + 0); //另一種呼叫圖片中畫素RGB值的方法
            G = inImg.at<uchar>(i,inImg.channels()*j + 1);
            R = inImg.at<uchar>(i,inImg.channels()*j + 2);*/

            int maxValue = max(max(B,G),R);
            int minValue = min(min(B,G),R);

            double S = (1-3.0*minValue/(R+G+B));

            //R > RT  R>=G>=B  S>=((255-R)*ST/RT)  
            if(R > redThre && R >= G && G >= B && S >0.20 && S >((255 - R) * saturationTh/redThre))
            {
                fireImg.at<uchar>(i,j) = 255;
            }
            else
            {
                fireImg.at<uchar>(i,j) = 0;
            }
        }
    }

    dilate(fireImg,fireImg,Mat(5,5,CV_8UC1));
    imshow("fire",fireImg);
    DrawFire(inImg,fireImg);
    waitKey(0);
    return fireImg;
}

void DrawFire(Mat &inputImg,Mat foreImg)
{
    vector<vector<Point>> contours_set;//儲存輪廓提取後的點集及拓撲關係

    findContours(foreImg,contours_set,CV_RETR_EXTERNAL,CV_CHAIN_APPROX_NONE);

    Mat result0;
    Scalar holeColor;
    Scalar externalColor;

    vector<vector<Point> >::iterator iter = contours_set.begin() ;
    for(; iter!= contours_set.end(); )
    {
        Rect rect = boundingRect(*iter );
        float radius;
        Point2f center;
        minEnclosingCircle(*iter,center,radius);

        if (rect.area()> 0)
        {

            rectangle(inputImg,rect,Scalar(0,255,0));
            ++ iter;

        }
        else
        {
            iter = contours_set.erase(iter);
        }
    }

    imshow("showFire",inputImg);
    waitKey(0);
}
 

提示：若有錯誤不足，歡迎批評指正。該系列文章僅供參考，歡迎互相學習交流。

感謝你的閱讀，期待你的關注收藏點贊！