簡體   English   中英

OpenCV僅在大輪廓周圍繪制矩形?

[英]OpenCV draw rectangles around only large contours?

第一次發布時,希望我以正確的方式輸入代碼。

我正在嘗試檢測和計數視頻中的車輛,因此,如果您看下面的代碼,在閾值化和擴張之后,我會找到圖像的輪廓,然后使用drawContours和矩形在檢測到的輪廓周圍繪制一個框。

我試圖通過說矩形的面積是否不大於40,000來在drawContours / rectangle if語句上放置一個過濾器,所以不要繪制它。

現在,如果您看一下我附帶的圖片,您會看到在較大的矩形內繪制了一些矩形,而我並不想這樣做。 在此處輸入圖片說明 這些矩形的面積小於40,000,但由於某些原因而被繪制。

我本打算使用矩形來計算圖像上的汽車,但是如果這不是最好的方法,那么我可以提出建議。

謝謝。

using namespace cv;
using namespace std;

int main()
{
    VideoCapture TestVideo;                 //Declare video capture
    Mat frame;                              //declare Mat as frame to grab

    TestVideo.open("FroggerHighway.mp4");           //open the test video from the project directory

    if (!TestVideo.isOpened())              //If its not open declare the error
        {
            cout << "Video did not open." << endl;
            waitKey(0);
        }

    if (TestVideo.get(CV_CAP_PROP_FRAME_COUNT) < 1)     //If the frame count is less than 1, basically an error checker
        {
            cout << "Video file must have at least one frame." << endl;
            waitKey(0);
        }

    TestVideo.read(frame);                              //read the first frame
    Mat frameGray = Mat::zeros(frame.size(), CV_8UC1);  //Convert frame source to gray
    cvtColor(frame, frameGray, CV_BGR2GRAY); 

    Mat frame2 = Mat::zeros(frameGray.size(), frameGray.type());    //Intermediate frame
    Mat framediff;                                                  //Frame differencing
    Mat thresh;
    Mat element;                                                    //Element used for morphOps (dilation)
    Mat dil;    

    while (TestVideo.isOpened() & waitKey(30) != 27)                //while the video is open, show the frame, press escape to end video
        {
        absdiff(frameGray, frame2, framediff);                      //take absolute difference of both frames
        threshold(framediff, thresh, 22, 255, CV_THRESH_BINARY);    //If absdiff is greater than 22, turn it white. 

        namedWindow("Gray", CV_WINDOW_NORMAL);                      //Display gray video
        imshow("Gray", frameGray);

        namedWindow("FrameDiff", CV_WINDOW_NORMAL);                 //Show frame difference before threshold/dilation
        imshow("FrameDiff", framediff);

        namedWindow("Threshold", CV_WINDOW_NORMAL);                 //Show thresholded video
        imshow("Threshold", thresh);

        element = getStructuringElement(MORPH_CROSS,                //morphOps dilation
            Size(2 * 5 + 1, 2 * 5 + 1),
            Point(5, 5));
        dilate(thresh, dil, element, Point(-1, -1), 1, 1, 1);

        namedWindow("Dilation", CV_WINDOW_NORMAL);                  //Show dilated video. 
        imshow("Dilation", dil);

        //Apply findCountours function to draw countours and count the objects.
        vector<vector<Point> > contours;        //Not sure what this does but it works
        findContours(dil, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);    //(outout image, hierarchy, and 2 ways to calculate it)
        vector<vector<Point> > contours_poly(contours.size());      //Also not sure what this does
        vector<Rect> boundRect(contours.size());            //This is used to approximate a polygon to fit the contours it calculated I think

        Mat output = Mat::zeros(dil.rows, dil.cols, CV_8UC3);   
        int counter = 0;                                    //Used to count # of rectangle drawn

        for (int i = 0; i < contours.size(); i++)
        {
            approxPolyDP(Mat(contours[i]), contours_poly[i], 3, true);  //Approximates a polygon to fit the contours calculated ?
            boundRect[i] = boundingRect(Mat(contours_poly[i]));    //for each approximation, a bounding rectangle is sorted around the contour ?

            if ((boundRect[i].x * boundRect[i].y) > 40000)          //If the bounding rectangle has an area less than 40,000 then just ignore it completely
            {
                counter = counter + 1;
                drawContours(output, contours, i, Scalar(255, 255, 255), -3);   //(input, countors, contour to be drawn, color of it, thickness (negative fills image));
                rectangle(output, boundRect[i].tl(), boundRect[i].br(), Scalar(0, 0, 255), 2, 8, 0); //Draws the actual rectangle around the contours
            }
        }

        cout << "Rectangles Drawn: " << counter << endl;

        namedWindow("Output", CV_WINDOW_NORMAL);
        imshow("Output", output);

            if (((TestVideo.get(CV_CAP_PROP_POS_FRAMES) + 1) < TestVideo.get(CV_CAP_PROP_FRAME_COUNT)) & (waitKey(30) != 27))   //Move the frame count up 1, show the frame
            {
                TestVideo.read(frame);
                frameGray.copyTo(frame2);                   //MUST USE copyTo, or clone! Can't do frame2 = frameGray*
                cvtColor(frame, frameGray, CV_BGR2GRAY);
            }
            else
            {
                cout << "End of Video" << endl;
                waitKey(0);
                break;
            }

            waitKey(30);        //wait 30ms between showing each frame
        }

    return (0);
}

您將xy坐標相乘以獲得矩形的面積,應該將widthheight相乘。

//If the bounding rectangle has an area less than 40,000 then just ignore it completely
if ((boundRect[i].width * boundRect[i].height) > 40000)          
{
// code here
}

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM